Typing (#29947)

Author: ShaharNaveh

pandas/_libs/tslibs/offsets.pyx

@@ -1,6 +1,7 @@
 import cython
 
 import time
+from typing import Any
 from cpython.datetime cimport (PyDateTime_IMPORT,
                                PyDateTime_Check,
                                PyDelta_Check,
@@ -328,7 +329,7 @@ class _BaseOffset:
     def __setattr__(self, name, value):
         raise AttributeError("DateOffset objects are immutable.")
 
-    def __eq__(self, other) -> bool:
+    def __eq__(self, other: Any) -> 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) -> bool:
+    def __eq__(self, other: Any) -> 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 List, Optional, Tuple, Type
+from typing import Any, 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) -> bool:
+    def __eq__(self, other: Any) -> bool:
         """
         Check whether 'other' is equal to self.
 

pandas/core/dtypes/dtypes.py

@@ -764,7 +764,7 @@ def __hash__(self) -> int:
         # TODO: update this.
         return hash(str(self))
 
-    def __eq__(self, other) -> bool:
+    def __eq__(self, other: Any) -> bool:
         if isinstance(other, str):
             return other == self.name
 
@@ -903,7 +903,7 @@ def __hash__(self) -> int:
         # make myself hashable
         return hash(str(self))
 
-    def __eq__(self, other) -> bool:
+    def __eq__(self, other: Any) -> bool:
         if isinstance(other, str):
             return other == self.name or other == self.name.title()
 
@@ -1076,7 +1076,7 @@ def __hash__(self) -> int:
         # make myself hashable
         return hash(str(self))
 
-    def __eq__(self, other) -> bool:
+    def __eq__(self, other: Any) -> 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,6 +7,8 @@
 
 """
 
+from typing import Any
+
 from pandas.core.base import PandasObject
 
 from pandas.io.formats.printing import pprint_thing
@@ -71,7 +73,7 @@ def __radd__(self, other):
             other = list(other)
         return type(self)(other + list(self))
 
-    def __eq__(self, other) -> bool:
+    def __eq__(self, other: Any) -> bool:
         if isinstance(other, (tuple, FrozenList)):
             other = list(other)
         return super().__eq__(other)

pandas/io/pytables.py

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

pandas/io/stata.py

@@ -15,6 +15,7 @@
 import os
 import struct
 import sys
+from typing import Any
 import warnings
 
 from dateutil.relativedelta import relativedelta
@@ -857,7 +858,7 @@ def __str__(self) -> str:
     def __repr__(self) -> str:
         return f"{type(self)}({self})"
 
-    def __eq__(self, other) -> bool:
+    def __eq__(self, other: Any) -> bool:
         return (
             isinstance(other, type(self))
             and self.string == other.string

pandas/tseries/offsets.py

@@ -1,7 +1,7 @@
 from datetime import date, datetime, timedelta
 import functools
 import operator
-from typing import Optional
+from typing import Any, Optional
 
 from dateutil.easter import easter
 import numpy as np
@@ -2551,7 +2551,7 @@ def __add__(self, other):
                 f"the add operation between {self} and {other} will overflow"
             )
 
-    def __eq__(self, other) -> bool:
+    def __eq__(self, other: Any) -> bool:
         if isinstance(other, str):
             from pandas.tseries.frequencies import to_offset