Annotations to __eq__ function

Author: MomIsBestFriend

pandas/_libs/tslibs/offsets.pyx

@@ -1,7 +1,6 @@
 import cython
 
 import time
-from typing import Any
 from cpython.datetime cimport (PyDateTime_IMPORT,
                                PyDateTime_Check,
                                PyDelta_Check,
@@ -329,7 +328,7 @@ class _BaseOffset:
     def __setattr__(self, name, value):
         raise AttributeError("DateOffset objects are immutable.")
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other) -> bool:
         if isinstance(other, str):
             try:
                 # GH#23524 if to_offset fails, we are dealing with an

pandas/core/arrays/sparse/dtype.py

@@ -90,7 +90,7 @@ def __hash__(self):
         # __eq__, so we explicitly do it here.
         return super().__hash__()
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other) -> bool:
         # We have to override __eq__ to handle NA values in _metadata.
         # The base class does simple == checks, which fail for NA.
         if isinstance(other, str):

pandas/core/dtypes/base.py

@@ -1,5 +1,5 @@
 """Extend pandas with custom array types"""
-from typing import Any, List, Optional, Tuple, Type
+from typing import List, Optional, Tuple, Type
 
 import numpy as np
 
@@ -86,7 +86,7 @@ def __from_arrow__(
     def __str__(self) -> str:
         return self.name
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other) -> bool:
         """
         Check whether 'other' is equal to self.
 

pandas/core/dtypes/dtypes.py

@@ -370,7 +370,7 @@ def __hash__(self) -> int:
         # We *do* want to include the real self.ordered here
         return int(self._hash_categories(self.categories, self._ordered))
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other) -> bool:
         """
         Rules for CDT equality:
         1) Any CDT is equal to the string 'category'
@@ -765,7 +765,7 @@ def __hash__(self) -> int:
         # TODO: update this.
         return hash(str(self))
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other) -> bool:
         if isinstance(other, str):
             return other == self.name
 
@@ -904,7 +904,7 @@ def __hash__(self) -> int:
         # make myself hashable
         return hash(str(self))
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other) -> bool:
         if isinstance(other, str):
             return other == self.name or other == self.name.title()
 
@@ -1077,7 +1077,7 @@ def __hash__(self) -> int:
         # make myself hashable
         return hash(str(self))
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other) -> bool:
         if isinstance(other, str):
             return other.lower() in (self.name.lower(), str(self).lower())
         elif not isinstance(other, IntervalDtype):

pandas/core/indexes/frozen.py

@@ -7,7 +7,6 @@
 - .levels & .codes (FrozenNDArray)
 
 """
-from typing import Any
 import warnings
 
 import numpy as np
@@ -78,7 +77,7 @@ def __radd__(self, other):
             other = list(other)
         return self.__class__(other + list(self))
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other) -> bool:
         if isinstance(other, (tuple, FrozenList)):
             other = list(other)
         return super().__eq__(other)

pandas/io/pytables.py

@@ -1764,7 +1764,7 @@ def __repr__(self) -> str:
             )
         )
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other) -> bool:
         """ compare 2 col items """
         return all(
             getattr(self, a, None) == getattr(other, a, None)
@@ -2092,7 +2092,7 @@ def __repr__(self) -> str:
             )
         )
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other) -> bool:
         """ compare 2 col items """
         return all(
             getattr(self, a, None) == getattr(other, a, None)

pandas/io/stata.py

@@ -16,7 +16,6 @@
 import os
 import struct
 import sys
-from typing import Any
 import warnings
 
 from dateutil.relativedelta import relativedelta
@@ -860,7 +859,7 @@ def __repr__(self) -> str:
         # not perfect :-/
         return "{cls}({obj})".format(cls=self.__class__, obj=self)
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other) -> bool:
         return (
             isinstance(other, self.__class__)
             and self.string == other.string

pandas/tests/groupby/test_function.py

@@ -3,7 +3,6 @@
 from io import StringIO
 from itertools import product
 from string import ascii_lowercase
-from typing import Any
 
 import numpy as np
 import pytest
@@ -1231,7 +1230,7 @@ def __init__(self, msg="I will raise inside Cython"):
             super().__init__()
             self.msg = msg
 
-        def __eq__(self, other: Any):
+        def __eq__(self, other):
             # gets called in Cython to check that raising calls the method
             raise RaisingObjectException(self.msg)
 

pandas/tests/indexing/test_indexing.py

@@ -2,7 +2,6 @@
 
 from datetime import datetime
 import re
-from typing import Any
 import weakref
 
 import numpy as np
@@ -596,7 +595,7 @@ def __str__(self) -> str:
 
             __repr__ = __str__
 
-            def __eq__(self, other: Any) -> bool:
+            def __eq__(self, other) -> bool:
                 return self.value == other.value
 
             def view(self):

pandas/tests/scalar/timedelta/test_timedelta.py

@@ -1,7 +1,6 @@
 """ test the scalar Timedelta """
 from datetime import timedelta
 import re
-from typing import Any
 
 import numpy as np
 import pytest
@@ -134,7 +133,7 @@ def generic_result(self):
                 else:
                     return self.cmp_result
 
-            def __eq__(self, other: Any):
+            def __eq__(self, other):
                 return self.generic_result()
 
             def __gt__(self, other):

pandas/tests/scalar/timestamp/test_comparisons.py

@@ -1,6 +1,5 @@
 from datetime import datetime
 import operator
-from typing import Any
 
 import numpy as np
 import pytest
@@ -180,7 +179,7 @@ def __gt__(self, o):
         def __ge__(self, o):
             return True
 
-        def __eq__(self, other: Any) -> bool:
+        def __eq__(self, other) -> bool:
             return isinstance(other, Inf)
 
     inf = Inf()

pandas/tests/series/test_ufunc.py

@@ -1,6 +1,5 @@
 from collections import deque
 import string
-from typing import Any
 
 import numpy as np
 import pytest
@@ -283,7 +282,7 @@ def __add__(self, other):
             other = getattr(other, "value", other)
             return type(self)(self.value + other)
 
-        def __eq__(self, other: Any) -> bool:
+        def __eq__(self, other) -> bool:
             return type(other) is Thing and self.value == other.value
 
         def __repr__(self) -> str:

pandas/tests/test_algos.py

@@ -1,7 +1,6 @@
 from datetime import datetime
 from itertools import permutations
 import struct
-from typing import Any
 
 import numpy as np
 from numpy.random import RandomState
@@ -768,7 +767,7 @@ def test_same_object_is_in(self):
         # with similar behavior, then we at least should
         # fall back to usual python's behavior: "a in [a] == True"
         class LikeNan:
-            def __eq__(self, other: Any) -> bool:
+            def __eq__(self, other) -> bool:
                 return False
 
             def __hash__(self):

pandas/tseries/offsets.py

@@ -1,7 +1,7 @@
 from datetime import date, datetime, timedelta
 import functools
 import operator
-from typing import Any, Optional
+from typing import Optional
 
 from dateutil.easter import easter
 import numpy as np
@@ -2577,7 +2577,7 @@ def __add__(self, other):
                 "will overflow".format(self=self, other=other)
             )
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other) -> bool:
         if isinstance(other, str):
             from pandas.tseries.frequencies import to_offset