hotfix(id_generation): ignore attributes that fail hashing

ZettaAI · Dec 28, 2023 · 63bd41f · 63bd41f
1 parent 7fb1390
commit 63bd41f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/tests/unit/mazepa/test_id_generation.py b/tests/unit/mazepa/test_id_generation.py
@@ -51,6 +51,10 @@ def __init__(self, a):
     def method(self, b):
         return self.a * b
 
+    @property
+    def prop(self):
+        return self.a
+
 
 @taskable_operation_cls()
 @attrs.mutable

diff --git a/zetta_utils/mazepa/id_generation.py b/zetta_utils/mazepa/id_generation.py
@@ -53,7 +53,14 @@ 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.__class__, attribute_name, None)
+
+        if attrib is not None and isinstance(attrib, property):
+            _get_code_hash(attrib, _hash, _visited)  # type: ignore
+            continue
+
         attrib = getattr(fn, attribute_name)
+
         if callable(attrib):
             _get_code_hash(attrib, _hash, _visited)
         else: