Fixed autoeq in case of attributeerror

smarie · Nov 17, 2019 · 7a4750a · 7a4750a
1 parent 6f09f96
commit 7a4750a
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/autoclass/autoeq_.py b/autoclass/autoeq_.py
@@ -157,10 +157,14 @@ def __eq__(self, other):
         """
         Generated by @autoeq. Relies on the hardcoded list of field names and "getattr" (object) for the value.
         """
-        for att_name in selected_names:
-            if getattr(self, att_name) != getattr(other, att_name):
-                return False
-        return True
+        try:
+            for att_name in selected_names:
+                if getattr(self, att_name) != getattr(other, att_name):
+                    return False
+        except AttributeError:
+            return False
+        else:
+            return True
 
     return __eq__