[Units] Handle other not WikiArticle in WikiArticle.__eq__

Harmon758 · Jul 30, 2023 · 7465798 · 7465798
1 parent a372806
commit 7465798
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/units/wikis.py b/units/wikis.py
@@ -29,7 +29,10 @@ class WikiArticle(BaseModel):
     wiki: WikiInfo
 
     def __eq__(self, other):
-        return self.url == other.url
+        if isinstance(other, WikiArticle):
+            return self.url == other.url
+        else:
+            return NotImplemented
 
     def __hash__(self) -> int:
         return hash(self.url)