Adjust behaviour in #1481 to return NotImplemented

weaversam8 · weaversam8 · commit 9d35d1b5e151 · 2024-10-25T15:51:05.000-04:00
diff --git a/lark/grammar.py b/lark/grammar.py
@@ -16,12 +16,8 @@ def __init__(self, name: str) -> None:
         self.name = name
 
     def __eq__(self, other):
-        if other is None:
-            return False
         if not isinstance(other, Symbol):
-            raise NotImplementedError(
-                f"Comparing a Symbol with type {type(other).__name__} is not implemented"
-            )
+            return NotImplemented
         return self.is_term == other.is_term and self.name == other.name
 
     def __ne__(self, other):
diff --git a/tests/test_grammar.py b/tests/test_grammar.py
@@ -302,14 +302,6 @@ def test_symbol_eq(self):
 
         self.assertNotEqual(a, b)
 
-        a = dict()
-
-        self.assertRaisesRegex(
-            NotImplementedError,
-            r"Comparing a Symbol with type dict is not implemented",
-            lambda: a == b,
-        )
-
 
 if __name__ == '__main__':
     main()
