Fix make_hashable

kuznia-rdzeni · Feb 10, 2025 · 7ca844b · 7ca844b
1 parent cb42874
commit 7ca844b
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/transactron/utils/data_repr.py b/transactron/utils/data_repr.py
@@ -24,12 +24,16 @@ def layout_subset(layout: StructLayout, *, fields: set[str]) -> StructLayout:
 
 
 def make_hashable(val) -> Hashable:
-    if isinstance(val, Mapping):
-        return frozenset(((k, make_hashable(v)) for k, v in val.items()))
-    elif isinstance(val, Iterable):
-        return tuple(make_hashable(v) for v in val)
-    else:
+    try:
+        hash(val)
         return val
+    except TypeError:
+        if isinstance(val, Mapping):
+            return frozenset(((k, make_hashable(v)) for k, v in val.items()))
+        elif isinstance(val, Iterable):
+            return tuple(make_hashable(v) for v in val)
+        else:
+            raise
 
 
 def align_to_power_of_two(num: int, power: int) -> int: