Merge branch 'main' into TYP_io

Author: nafarya

doc/make.py

@@ -45,7 +45,7 @@ def __init__(
         single_doc=None,
         verbosity=0,
         warnings_are_errors=False,
-    ):
+    ) -> None:
         self.num_jobs = num_jobs
         self.include_api = include_api
         self.whatsnew = whatsnew

pandas/_config/config.py

@@ -204,7 +204,7 @@ def get_default_val(pat: str):
 class DictWrapper:
     """provide attribute-style access to a nested dict"""
 
-    def __init__(self, d: dict[str, Any], prefix: str = ""):
+    def __init__(self, d: dict[str, Any], prefix: str = "") -> None:
         object.__setattr__(self, "d", d)
         object.__setattr__(self, "prefix", prefix)
 
@@ -248,7 +248,7 @@ def __dir__(self) -> Iterable[str]:
 
 
 class CallableDynamicDoc:
-    def __init__(self, func, doc_tmpl):
+    def __init__(self, func, doc_tmpl) -> None:
         self.__doc_tmpl__ = doc_tmpl
         self.__func__ = func
 
@@ -422,7 +422,7 @@ class option_context(ContextDecorator):
     ...     pass
     """
 
-    def __init__(self, *args):
+    def __init__(self, *args) -> None:
         if len(args) % 2 != 0 or len(args) < 2:
             raise ValueError(
                 "Need to invoke as option_context(pat, val, [(pat, val), ...])."

pandas/_testing/contexts.py

@@ -228,7 +228,7 @@ class RNGContext:
         np.random.randn()
     """
 
-    def __init__(self, seed):
+    def __init__(self, seed) -> None:
         self.seed = seed
 
     def __enter__(self):

pandas/compat/numpy/function.py

@@ -40,7 +40,7 @@ def __init__(
         fname=None,
         method: str | None = None,
         max_fname_arg_count=None,
-    ):
+    ) -> None:
         self.fname = fname
         self.method = method
         self.defaults = defaults

pandas/conftest.py

@@ -447,7 +447,7 @@ def dict_subclass():
     """
 
     class TestSubDict(dict):
-        def __init__(self, *args, **kwargs):
+        def __init__(self, *args, **kwargs) -> None:
             dict.__init__(self, *args, **kwargs)
 
     return TestSubDict
@@ -460,7 +460,7 @@ def non_dict_mapping_subclass():
     """
 
     class TestNonDictMapping(abc.Mapping):
-        def __init__(self, underlying_dict):
+        def __init__(self, underlying_dict) -> None:
             self._data = underlying_dict
 
         def __getitem__(self, key):
@@ -1709,7 +1709,7 @@ class TestMemoryFS(MemoryFileSystem):
         protocol = "testmem"
         test = [None]
 
-        def __init__(self, **kwargs):
+        def __init__(self, **kwargs) -> None:
             self.test[0] = kwargs.pop("test", None)
             super().__init__(**kwargs)
 

pandas/core/generic.py

@@ -2956,7 +2956,7 @@ def to_sql(
     )
     def to_pickle(
         self,
-        path,
+        path: FilePath | WriteBuffer[bytes],
         compression: CompressionOptions = "infer",
         protocol: int = pickle.HIGHEST_PROTOCOL,
         storage_options: StorageOptions = None,
@@ -2966,8 +2966,10 @@ def to_pickle(
 
         Parameters
         ----------
-        path : str
-            File path where the pickled object will be stored.
+        path : str, path object, or file-like object
+            String, path object (implementing ``os.PathLike[str]``), or file-like
+            object implementing a binary ``write()`` function. File path where
+            the pickled object will be stored.
         {compression_options}
         protocol : int
             Int which indicates which protocol should be used by the pickler,

pandas/core/indexes/multi.py

@@ -1088,9 +1088,7 @@ def _engine(self):
         # equivalent to sorting lexicographically the codes themselves. Notice
         # that each level needs to be shifted by the number of bits needed to
         # represent the _previous_ ones:
-        offsets = np.concatenate([lev_bits[1:], [0]]).astype(  # type: ignore[arg-type]
-            "uint64"
-        )
+        offsets = np.concatenate([lev_bits[1:], [0]]).astype("uint64")
 
         # Check the total number of bits needed for our representation:
         if lev_bits[0] > 64:

pandas/errors/__init__.py

@@ -186,7 +186,7 @@ class AbstractMethodError(NotImplementedError):
     while keeping compatibility with Python 2 and Python 3.
     """
 
-    def __init__(self, class_instance, methodtype="method"):
+    def __init__(self, class_instance, methodtype="method") -> None:
         types = {"method", "classmethod", "staticmethod", "property"}
         if methodtype not in types:
             raise ValueError(

pandas/tests/base/test_constructors.py

@@ -51,7 +51,7 @@ def bar(self, *args, **kwargs):
             pass
 
     class Delegate(PandasDelegate, PandasObject):
-        def __init__(self, obj):
+        def __init__(self, obj) -> None:
             self.obj = obj
 
     def test_invalid_delegation(self):

pandas/tests/dtypes/test_inference.py

@@ -91,7 +91,7 @@ class MockNumpyLikeArray:
     a scalar (`is_scalar(np.array(1)) == False`), but it is not list-like either.
     """
 
-    def __init__(self, values):
+    def __init__(self, values) -> None:
         self._values = values
 
     def __iter__(self):
@@ -323,7 +323,7 @@ def test_is_dict_like_fails(ll):
 @pytest.mark.parametrize("has_contains", [True, False])
 def test_is_dict_like_duck_type(has_keys, has_getitem, has_contains):
     class DictLike:
-        def __init__(self, d):
+        def __init__(self, d) -> None:
             self.d = d
 
         if has_keys:
@@ -1937,7 +1937,7 @@ def test_is_scalar_number(self):
         #  subclasses are.
 
         class Numeric(Number):
-            def __init__(self, value):
+            def __init__(self, value) -> None:
                 self.value = value
 
             def __int__(self):

pandas/tests/extension/arrow/arrays.py

@@ -179,7 +179,7 @@ def all(self, axis=0, out=None):
 
 
 class ArrowBoolArray(ArrowExtensionArray):
-    def __init__(self, values):
+    def __init__(self, values) -> None:
         if not isinstance(values, pa.ChunkedArray):
             raise ValueError
 
@@ -189,7 +189,7 @@ def __init__(self, values):
 
 
 class ArrowStringArray(ArrowExtensionArray):
-    def __init__(self, values):
+    def __init__(self, values) -> None:
         if not isinstance(values, pa.ChunkedArray):
             raise ValueError
 

pandas/tests/extension/arrow/test_timestamp.py

@@ -40,7 +40,7 @@ def construct_array_type(cls) -> type_t[ArrowTimestampUSArray]:
 
 
 class ArrowTimestampUSArray(ArrowExtensionArray):
-    def __init__(self, values):
+    def __init__(self, values) -> None:
         if not isinstance(values, pa.ChunkedArray):
             raise ValueError
 

pandas/tests/extension/decimal/array.py

@@ -41,7 +41,7 @@ class DecimalDtype(ExtensionDtype):
     na_value = decimal.Decimal("NaN")
     _metadata = ("context",)
 
-    def __init__(self, context=None):
+    def __init__(self, context=None) -> None:
         self.context = context or decimal.getcontext()
 
     def __repr__(self) -> str:
@@ -66,7 +66,7 @@ def _is_numeric(self) -> bool:
 class DecimalArray(OpsMixin, ExtensionScalarOpsMixin, ExtensionArray):
     __array_priority__ = 1000
 
-    def __init__(self, values, dtype=None, copy=False, context=None):
+    def __init__(self, values, dtype=None, copy=False, context=None) -> None:
         for i, val in enumerate(values):
             if is_float(val):
                 if np.isnan(val):

pandas/tests/extension/json/array.py

@@ -67,7 +67,7 @@ class JSONArray(ExtensionArray):
     dtype = JSONDtype()
     __array_priority__ = 1000
 
-    def __init__(self, values, dtype=None, copy=False):
+    def __init__(self, values, dtype=None, copy=False) -> None:
         for val in values:
             if not isinstance(val, self.dtype.type):
                 raise TypeError("All values must be of type " + str(self.dtype.type))

pandas/tests/extension/list/array.py

@@ -44,7 +44,7 @@ class ListArray(ExtensionArray):
     dtype = ListDtype()
     __array_priority__ = 1000
 
-    def __init__(self, values, dtype=None, copy=False):
+    def __init__(self, values, dtype=None, copy=False) -> None:
         if not isinstance(values, np.ndarray):
             raise TypeError("Need to pass a numpy array as values")
         for val in values:

pandas/tests/extension/test_common.py

@@ -14,7 +14,7 @@ class DummyDtype(dtypes.ExtensionDtype):
 
 
 class DummyArray(ExtensionArray):
-    def __init__(self, data):
+    def __init__(self, data) -> None:
         self.data = data
 
     def __array__(self, dtype):

pandas/tests/extension/test_extension.py

@@ -8,7 +8,7 @@
 
 
 class MyEA(ExtensionArray):
-    def __init__(self, values):
+    def __init__(self, values) -> None:
         self._values = values
 
 

pandas/tests/frame/constructors/test_from_records.py

@@ -194,7 +194,7 @@ def test_from_records_bad_index_column(self):
 
     def test_from_records_non_tuple(self):
         class Record:
-            def __init__(self, *args):
+            def __init__(self, *args) -> None:
                 self.args = args
 
             def __getitem__(self, i):

pandas/tests/frame/methods/test_select_dtypes.py

@@ -15,7 +15,7 @@
 class DummyDtype(ExtensionDtype):
     type = int
 
-    def __init__(self, numeric):
+    def __init__(self, numeric) -> None:
         self._numeric = numeric
 
     @property
@@ -28,7 +28,7 @@ def _is_numeric(self):
 
 
 class DummyArray(ExtensionArray):
-    def __init__(self, data, dtype):
+    def __init__(self, data, dtype) -> None:
         self.data = data
         self._dtype = dtype
 

pandas/tests/frame/methods/test_set_index.py

@@ -595,7 +595,7 @@ def test_set_index_custom_label_type(self):
         # GH#24969
 
         class Thing:
-            def __init__(self, name, color):
+            def __init__(self, name, color) -> None:
                 self.name = name
                 self.color = color
 
@@ -673,7 +673,7 @@ def test_set_index_custom_label_type_raises(self):
 
         # purposefully inherit from something unhashable
         class Thing(set):
-            def __init__(self, name, color):
+            def __init__(self, name, color) -> None:
                 self.name = name
                 self.color = color
 

pandas/tests/frame/methods/test_to_records.py

@@ -358,7 +358,7 @@ def test_to_records_dtype_mi(self, df, kwargs, expected):
     def test_to_records_dict_like(self):
         # see GH#18146
         class DictLike:
-            def __init__(self, **kwargs):
+            def __init__(self, **kwargs) -> None:
                 self.d = kwargs.copy()
 
             def __getitem__(self, key):

pandas/tests/frame/test_arithmetic.py

@@ -41,7 +41,7 @@ def switch_numexpr_min_elements(request):
 
 
 class DummyElement:
-    def __init__(self, value, dtype):
+    def __init__(self, value, dtype) -> None:
         self.value = value
         self.dtype = np.dtype(dtype)
 

pandas/tests/frame/test_constructors.py

@@ -1363,7 +1363,7 @@ def test_constructor_sequence_like(self):
         # collections.Sequence like
 
         class DummyContainer(abc.Sequence):
-            def __init__(self, lst):
+            def __init__(self, lst) -> None:
                 self._lst = lst
 
             def __getitem__(self, n):

pandas/tests/frame/test_stack_unstack.py

@@ -1861,7 +1861,7 @@ def test_unstack_number_of_levels_larger_than_int32(self, monkeypatch):
         # GH 26314: Change ValueError to PerformanceWarning
 
         class MockUnstacker(reshape_lib._Unstacker):
-            def __init__(self, *args, **kwargs):
+            def __init__(self, *args, **kwargs) -> None:
                 # __init__ will raise the warning
                 super().__init__(*args, **kwargs)
                 raise Exception("Don't compute final result.")

pandas/tests/frame/test_subclass.py

@@ -42,7 +42,7 @@ class CustomDataFrame(DataFrame):
             custom plotting functions.
             """
 
-            def __init__(self, *args, **kw):
+            def __init__(self, *args, **kw) -> None:
                 super().__init__(*args, **kw)
 
             @property

pandas/tests/frame/test_ufunc.py

@@ -263,12 +263,6 @@ def test_alignment_deprecation_many_inputs(request):
     )
 
     if np_version_gte1p22:
-        mark = pytest.mark.xfail(
-            reason="ufunc 'my_ufunc' did not contain a loop with signature matching "
-            "types",
-        )
-        request.node.add_marker(mark)
-
         mark = pytest.mark.filterwarnings(
             "ignore:`np.MachAr` is deprecated.*:DeprecationWarning"
         )

pandas/tests/groupby/test_counting.py

@@ -364,7 +364,7 @@ class RaisingObjectException(Exception):
         pass
 
     class RaisingObject:
-        def __init__(self, msg="I will raise inside Cython"):
+        def __init__(self, msg="I will raise inside Cython") -> None:
             super().__init__()
             self.msg = msg
 

pandas/tests/indexes/datetimes/test_timezones.py

@@ -40,7 +40,7 @@
 class FixedOffset(tzinfo):
     """Fixed offset in minutes east from UTC."""
 
-    def __init__(self, offset, name):
+    def __init__(self, offset, name) -> None:
         self.__offset = timedelta(minutes=offset)
         self.__name = name
 

pandas/tests/indexes/test_index_new.py

@@ -352,7 +352,7 @@ def test_constructor_ndarray_like(self, array):
         # it should be possible to convert any object that satisfies the numpy
         # ndarray interface directly into an Index
         class ArrayLike:
-            def __init__(self, array):
+            def __init__(self, array) -> None:
                 self.array = array
 
             def __array__(self, dtype=None) -> np.ndarray:

pandas/tests/indexing/test_iloc.py

@@ -1072,7 +1072,7 @@ def test_iloc_getitem_float_duplicates(self):
     def test_iloc_setitem_custom_object(self):
         # iloc with an object
         class TO:
-            def __init__(self, value):
+            def __init__(self, value) -> None:
                 self.value = value
 
             def __str__(self) -> str:

pandas/tests/io/formats/test_console.py

@@ -12,7 +12,7 @@ class MockEncoding:
     side effect should be an exception that will be raised.
     """
 
-    def __init__(self, encoding):
+    def __init__(self, encoding) -> None:
         super().__init__()
         self.val = encoding