hotfix(id_generation): ignore attributes that fail hashing

ZettaAI · Dec 28, 2023 · 0b011ea · 0b011ea
1 parent 7fb1390
commit 0b011ea
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/zetta_utils/mazepa/id_generation.py b/zetta_utils/mazepa/id_generation.py
@@ -53,7 +53,11 @@ def _get_code_hash(
     _visited.add(id(fn))
 
     for attribute_name in {x for x in dir(fn) if not x.startswith("__")}:
-        attrib = getattr(fn, attribute_name)
+        try:
+            attrib = getattr(fn, attribute_name)
+        except AssertionError:
+            # Might happen before proper initialization
+            continue
         if callable(attrib):
             _get_code_hash(attrib, _hash, _visited)
         else: