Fix __eq__ for LayerAttributes

KULeuven-MICAS · May 15, 2024 · 457fd33 · 457fd33
1 parent d417d51
commit 457fd33
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/zigzag/workload/LayerAttribute.py b/zigzag/workload/LayerAttribute.py
@@ -30,3 +30,6 @@ def __repr__(self):
 
     def __jsonrepr__(self) -> Any:
         return json_repr_handler(self.data)
+
+    def __eq__(self, other: object):
+        return isinstance(other, LayerAttribute) and self.data == other.data