Merge branch 'main' of github.com:loicdiridollou/pandas-stubs into ghxxx_cleanup_ann

Author: loicdiridollou

.pre-commit-config.yaml

@@ -3,7 +3,7 @@ ci:
     autofix_prs: false
 repos:
 -   repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.14.3
+    rev: v0.14.9
     hooks:
     -   id: ruff-check
         args: [--exit-non-zero-on-fix]
@@ -18,6 +18,6 @@ repos:
     hooks:
     -   id: isort
 -   repo: https://github.com/psf/black
-    rev: 25.9.0
+    rev: 25.12.0
     hooks:
     -   id: black

pandas-stubs/_typing.pyi

@@ -959,7 +959,10 @@ np_1darray_dt: TypeAlias = np_1darray[np.datetime64]
 np_1darray_td: TypeAlias = np_1darray[np.timedelta64]
 np_2darray: TypeAlias = np.ndarray[tuple[int, int], np.dtype[GenericT]]
 
-NDArrayT = TypeVar("NDArrayT", bound=np.ndarray)
+if sys.version_info >= (3, 11):
+    NDArrayT = TypeVar("NDArrayT", bound=np.ndarray)
+else:
+    NDArrayT = TypeVar("NDArrayT", bound=np.ndarray[Any, Any])
 
 DtypeNp = TypeVar("DtypeNp", bound=np.dtype[np.generic])
 KeysArgType: TypeAlias = Any

pandas-stubs/core/base.pyi

@@ -1,6 +1,4 @@
 from collections.abc import (
-    Callable,
-    Hashable,
     Iterator,
     Sequence,
 )
@@ -12,7 +10,6 @@ from typing import (
     Protocol,
     TypeAlias,
     TypeVar,
-    final,
     overload,
     type_check_only,
 )
@@ -43,7 +40,6 @@ from pandas._typing import (
     GenericT_co,
     Just,
     ListLike,
-    NDFrameT,
     Scalar,
     SupportsDType,
     np_1darray,
@@ -55,22 +51,12 @@ from pandas._typing import (
     np_ndarray_float,
     np_ndarray_td,
 )
-from pandas.util._decorators import cache_readonly
 
 T_INTERVAL_NP = TypeVar("T_INTERVAL_NP", bound=np.bytes_ | np.str_)
 
 class NoNewAttributesMixin:
     def __setattr__(self, key: str, value: Any) -> None: ...
 
-class SelectionMixin(Generic[NDFrameT]):
-    obj: NDFrameT
-    exclusions: frozenset[Hashable]
-    @final
-    @cache_readonly
-    def ndim(self) -> int: ...
-    def __getitem__(self, key: Any) -> Any: ...
-    def aggregate(self, func: Callable[..., Any], *args: Any, **kwargs: Any) -> Any: ...
-
 class IndexOpsMixin(OpsMixin, Generic[S1, GenericT_co]):
     __array_priority__: int = ...
     @property

pandas-stubs/core/frame.pyi

@@ -340,28 +340,23 @@ class _AtIndexerFrame(_AtIndexer):
         value: Scalar | NAType | NaTType | None,
     ) -> None: ...
 
-# With mypy 1.14.1 and python 3.12, the second overload needs a type-ignore statement
-if sys.version_info >= (3, 12):
-    class _GetItemHack:
-        @overload
-        def __getitem__(self, key: Scalar | tuple[Hashable, ...]) -> Series: ...  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
+class _GetItemHack:
+    @overload
+    def __getitem__(self, key: Scalar | tuple[Hashable, ...]) -> Series: ...  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
+    # With python 3.12+, the second overload needs a type-ignore statement
+    if sys.version_info >= (3, 12):
         @overload
         def __getitem__(  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
             self, key: Iterable[Hashable] | slice
         ) -> Self: ...
-        @overload
-        def __getitem__(self, key: Hashable) -> Series: ...
-
-else:
-    class _GetItemHack:
-        @overload
-        def __getitem__(self, key: Scalar | tuple[Hashable, ...]) -> Series: ...  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
+    else:
         @overload
         def __getitem__(  # pyright: ignore[reportOverlappingOverload]
             self, key: Iterable[Hashable] | slice
         ) -> Self: ...
-        @overload
-        def __getitem__(self, key: Hashable) -> Series: ...
+
+    @overload
+    def __getitem__(self, key: Hashable) -> Series: ...
 
 _AstypeArgExt: TypeAlias = (
     AstypeArg
@@ -562,16 +557,29 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
         coerce_float: bool = False,
         nrows: int | None = None,
     ) -> Self: ...
-    def to_records(
-        self,
-        index: _bool = True,
-        column_dtypes: (
-            _str | npt.DTypeLike | Mapping[HashableT1, npt.DTypeLike] | None
-        ) = None,
-        index_dtypes: (
-            _str | npt.DTypeLike | Mapping[HashableT2, npt.DTypeLike] | None
-        ) = None,
-    ) -> np.recarray: ...
+    if sys.version_info >= (3, 11):
+        def to_records(
+            self,
+            index: _bool = True,
+            column_dtypes: (
+                _str | npt.DTypeLike | Mapping[HashableT1, npt.DTypeLike] | None
+            ) = None,
+            index_dtypes: (
+                _str | npt.DTypeLike | Mapping[HashableT2, npt.DTypeLike] | None
+            ) = None,
+        ) -> np.recarray: ...
+    else:
+        def to_records(
+            self,
+            index: _bool = True,
+            column_dtypes: (
+                _str | npt.DTypeLike | Mapping[HashableT1, npt.DTypeLike] | None
+            ) = None,
+            index_dtypes: (
+                _str | npt.DTypeLike | Mapping[HashableT2, npt.DTypeLike] | None
+            ) = None,
+        ) -> np.recarray[Any, Any]: ...
+
     @overload
     def to_stata(
         self,

pandas-stubs/core/groupby/groupby.pyi

@@ -18,7 +18,6 @@ from typing import (
 )
 
 import numpy as np
-from pandas.core.base import SelectionMixin
 from pandas.core.frame import DataFrame
 from pandas.core.groupby import generic
 from pandas.core.groupby.indexing import (
@@ -350,7 +349,7 @@ class GroupByPlot(PlotAccessor, Generic[_GroupByT]):
     # def __call__(self, *args: Any, **kwargs: Any): ...
     # def __getattr__(self, name: str): ...
 
-class BaseGroupBy(SelectionMixin[NDFrameT], GroupByIndexingMixin):
+class BaseGroupBy(GroupByIndexingMixin, Generic[NDFrameT]):
     @final
     def __len__(self) -> int: ...
     @final

pandas-stubs/core/groupby/ops.pyi

@@ -1,6 +1,4 @@
-from collections.abc import (
-    Iterator,
-)
+from collections.abc import Iterator
 from typing import Generic
 
 from pandas._typing import (

pandas-stubs/core/indexes/multi.pyi

@@ -144,9 +144,7 @@ class MultiIndex(Index):
     @overload
     def unique(  # ty: ignore[invalid-method-override]  # pyright: ignore[reportIncompatibleMethodOverride]
         self, level: Level
-    ) -> (
-        Index
-    ): ...  # ty: ignore[invalid-method-override] # pyrefly: ignore[bad-override]
+    ) -> Index: ...
     def to_frame(  # type: ignore[override] # pyright: ignore[reportIncompatibleMethodOverride]
         self,
         index: bool = True,
@@ -174,7 +172,7 @@ class MultiIndex(Index):
     @overload
     def append(  # pyright: ignore[reportIncompatibleMethodOverride]
         self, other: Index | Sequence[Index]
-    ) -> Index: ...  # pyrefly: ignore[bad-override]
+    ) -> Index: ...
     def drop(self, codes: Level | Sequence[Level], level: Level | None = None, errors: str = "raise") -> MultiIndex: ...  # type: ignore[override] # pyright: ignore[reportIncompatibleMethodOverride]
     def swaplevel(self, i: int = -2, j: int = -1) -> Self: ...
     def reorder_levels(self, order: Sequence[Level]) -> MultiIndex: ...

pandas-stubs/core/indexes/range.pyi

@@ -66,7 +66,7 @@ class RangeIndex(_IndexSubclassBase[int, np.int64]):
     def is_monotonic_decreasing(self) -> bool: ...
     @property
     def has_duplicates(self) -> bool: ...
-    def factorize(  # ty: ignore[invalid-method-override]
+    def factorize(
         self, sort: bool = False, use_na_sentinel: bool = True
     ) -> tuple[np_1darray_intp, RangeIndex]: ...
     @property

pandas-stubs/core/series.pyi

@@ -563,11 +563,11 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
         self, ufunc: Callable[..., Any], method: _str, *inputs: Any, **kwargs: Any
     ) -> Any: ...
     if sys.version_info >= (3, 11):
-        def __array__(  # ty: ignore[invalid-method-override]
+        def __array__(
             self, dtype: _str | np.dtype = ..., copy: bool | None = ...
         ) -> np_1darray: ...
     else:
-        def __array__(  # ty: ignore[invalid-method-override]
+        def __array__(
             self, dtype: _str | np.dtype[Any] = ..., copy: bool | None = ...
         ) -> np_1darray: ...
 

pandas-stubs/core/window/ewm.pyi

@@ -1,5 +1,3 @@
-from typing import Any
-
 from pandas import (
     DataFrame,
     Series,
@@ -47,29 +45,3 @@ class ExponentialMovingWindow(BaseWindow[NDFrameT]):
 class ExponentialMovingWindowGroupby(
     BaseWindowGroupby[NDFrameT], ExponentialMovingWindow[NDFrameT]
 ): ...
-
-class OnlineExponentialMovingWindow(ExponentialMovingWindow[NDFrameT]):
-    def reset(self) -> None: ...
-    def aggregate(self, func, *args: Any, **kwargs: Any): ...
-    def std(self, bias: bool = False, *args: Any, **kwargs: Any): ...
-    def corr(
-        self,
-        other: DataFrame | Series | None = None,
-        pairwise: bool | None = None,
-        numeric_only: bool = False,
-    ): ...
-    def cov(
-        self,
-        other: DataFrame | Series | None = None,
-        pairwise: bool | None = None,
-        bias: bool = False,
-        numeric_only: bool = False,
-    ): ...
-    def var(self, bias: bool = False, numeric_only: bool = False): ...
-    def mean(
-        self,
-        *args: Any,
-        update: NDFrameT | None = ...,
-        update_times: None = None,
-        **kwargs: Any,
-    ) -> NDFrameT: ...