Merge branch 'main' into gh1327_date_timedelta_range

Author: loicdiridollou

pandas-stubs/_libs/tslibs/offsets.pyi

@@ -29,11 +29,11 @@ class ApplyTypeError(TypeError): ...
 class BaseOffset:
     n: int
     def __init__(self, n: int = ..., normalize: bool = ...) -> None: ...
-    def __eq__(self, other) -> bool: ...
-    def __ne__(self, other) -> bool: ...
+    def __eq__(self, other: object) -> bool: ...
+    def __ne__(self, other: object) -> bool: ...
     def __hash__(self) -> int: ...
     @property
-    def kwds(self) -> dict: ...
+    def kwds(self) -> dict[str, Any]: ...
     @property
     def base(self) -> BaseOffset: ...
     @overload
@@ -67,7 +67,6 @@ class BaseOffset:
     def __rsub__(self, other: BaseOffset) -> Self: ...
     @overload
     def __rsub__(self, other: _TimedeltaT) -> _TimedeltaT: ...
-    def __call__(self, other): ...
     @overload
     def __mul__(self, other: np.ndarray) -> np.ndarray: ...
     @overload
@@ -87,8 +86,6 @@ class BaseOffset:
     def rollback(self, dt: datetime) -> datetime: ...
     def rollforward(self, dt: datetime) -> datetime: ...
     def is_on_offset(self, dt: datetime) -> bool: ...
-    def __setstate__(self, state) -> None: ...
-    def __getstate__(self): ...
     @property
     def nanos(self) -> int: ...
 

pandas-stubs/_typing.pyi

@@ -15,6 +15,7 @@ from re import Pattern
 import sys
 from typing import (
     Any,
+    Generic,
     Literal,
     Protocol,
     SupportsIndex,
@@ -37,6 +38,7 @@ from typing_extensions import (
     ParamSpec,
     TypeAlias,
     TypeVar,
+    override,
 )
 
 from pandas._libs.interval import Interval
@@ -818,6 +820,8 @@ np_ndarray_float: TypeAlias = npt.NDArray[np.floating]
 np_ndarray_complex: TypeAlias = npt.NDArray[np.complexfloating]
 np_ndarray_bool: TypeAlias = npt.NDArray[np.bool_]
 np_ndarray_str: TypeAlias = npt.NDArray[np.str_]
+np_ndarray_dt: TypeAlias = npt.NDArray[np.datetime64]
+np_ndarray_td: TypeAlias = npt.NDArray[np.timedelta64]
 
 # Define shape and generic type variables with defaults similar to numpy
 GenericT = TypeVar("GenericT", bound=np.generic, default=Any)
@@ -1055,4 +1059,14 @@ DictConvertible: TypeAlias = FulldatetimeDict | DataFrame
 # where it is the only acceptable type.
 Incomplete: TypeAlias = Any
 
+# differentiating between bool and int/float/complex
+# https://github.com/pandas-dev/pandas-stubs/pull/1312#pullrequestreview-3126128971
+class Just(Protocol, Generic[T]):
+    @property  # type: ignore[override]
+    @override
+    def __class__(self, /) -> type[T]: ...
+    @__class__.setter
+    @override
+    def __class__(self, t: type[T], /) -> None: ...
+
 __all__ = ["npt", "type_t"]

pandas-stubs/core/arrays/base.pyi

@@ -1,3 +1,4 @@
+from collections.abc import Iterator
 from typing import (
     Any,
     overload,
@@ -23,16 +24,16 @@ class ExtensionArray:
     def __getitem__(self, item: ScalarIndexer) -> Any: ...
     @overload
     def __getitem__(self, item: SequenceIndexer) -> Self: ...
-    def __setitem__(self, key: int | slice | np.ndarray, value) -> None: ...
+    def __setitem__(self, key: int | slice | np.ndarray, value: Any) -> None: ...
     def __len__(self) -> int: ...
-    def __iter__(self): ...
+    def __iter__(self) -> Iterator[Any]: ...
     def __contains__(self, item: object) -> bool | np.bool_: ...
     def to_numpy(
         self,
         dtype: npt.DTypeLike | None = ...,
         copy: bool = False,
         na_value: Scalar = ...,
-    ) -> np_1darray: ...
+    ) -> np_1darray[Any]: ...
     @property
     def dtype(self) -> ExtensionDtype: ...
     @property

pandas-stubs/core/groupby/groupby.pyi

@@ -74,9 +74,9 @@ from pandas._typing import (
 from pandas.plotting import PlotAccessor
 
 _ResamplerGroupBy: TypeAlias = (
-    DatetimeIndexResamplerGroupby[NDFrameT]  # ty: ignore
-    | PeriodIndexResamplerGroupby[NDFrameT]  # ty: ignore
-    | TimedeltaIndexResamplerGroupby[NDFrameT]  # ty: ignore
+    DatetimeIndexResamplerGroupby[NDFrameT]  # ty: ignore[invalid-argument-type]
+    | PeriodIndexResamplerGroupby[NDFrameT]  # ty: ignore[invalid-argument-type]
+    | TimedeltaIndexResamplerGroupby[NDFrameT]  # ty: ignore[invalid-argument-type]
 )
 
 class GroupBy(BaseGroupBy[NDFrameT]):