👔 Handle import time case for __eq__

Signed-off-by: Gaurav-Kumbhat <[email protected]>
IBM · Aug 12, 2022 · 192f78e · 192f78e
1 parent e94db07
commit 192f78e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/import_tracker/lazy_import_errors.py b/import_tracker/lazy_import_errors.py
@@ -241,9 +241,10 @@ def __delete__(self, *_, **__):
     def __delitem__(self, *_, **__):
         self._raise()
 
-    def __eq__(self, *_, **__):
+    def __eq__(self, other, *_, **__):
         if not _is_import_time():
             self._raise()
+        return id(self) == id(other)
 
     def __float__(self, *_, **__):
         self._raise()