make CI pass

Author: twoertwein

pandas-stubs/core/arrays/base.pyi

@@ -10,11 +10,11 @@ from pandas._typing import (
     npt,
 )
 
-from pandas.core.dtypes.dtypes import ExtensionDtype as ExtensionDtype
+from pandas.core.dtypes.dtypes import ExtensionDtype
 
 class ExtensionArray:
     def __getitem__(self, item) -> Any: ...
-    def __setitem__(self, key: int | slice | np.ndarray, value) -> None: ...
+    def __setitem__(self, key: int | slice | np.ndarray, value: Self) -> None: ...
     def __len__(self) -> int: ...
     def __iter__(self): ...
     def __contains__(self, item: object) -> bool | np.bool_: ...

pandas-stubs/core/arrays/boolean.pyi

@@ -6,7 +6,7 @@ from typing import (
 
 import numpy as np
 from pandas.core.arrays import ExtensionArray
-from pandas.core.arrays.masked import BaseMaskedArray as BaseMaskedArray
+from pandas.core.arrays.masked import BaseMaskedArray
 from typing_extensions import TypeAlias
 
 from pandas._libs.missing import NAType
@@ -15,7 +15,7 @@ from pandas._typing import (
     type_t,
 )
 
-from pandas.core.dtypes.base import ExtensionDtype as ExtensionDtype
+from pandas.core.dtypes.base import ExtensionDtype
 
 class BooleanDtype(ExtensionDtype):
     na_value: ClassVar[NAType]
@@ -44,7 +44,7 @@ class BooleanArray(BaseMaskedArray):
         | npt.NDArray[np.bool_]
         | BooleanArray,
     ) -> None: ...
-    @overload  # type: ignore[override]
+    @overload
     def __getitem__(self, item: int) -> bool | NAType: ...
     @overload
     def __getitem__(self, item: _ArrayKey) -> BooleanArray: ...
@@ -55,4 +55,3 @@ class BooleanArray(BaseMaskedArray):
     ) -> np.ndarray | ExtensionArray: ...
     def any(self, skipna: bool = ..., **kwargs) -> bool: ...
     def all(self, skipna: bool = ..., **kwargs) -> bool: ...
-    def __setitem__(self, key, value) -> None: ...

pandas-stubs/core/arrays/integer.pyi

@@ -12,7 +12,6 @@ from pandas.arrays import (
     StringArray,
 )
 from pandas.core.arrays.masked import BaseMaskedArray
-from typing_extensions import Self
 
 from pandas._libs.missing import NAType
 from pandas._typing import npt
@@ -26,7 +25,7 @@ class _IntegerDtype(ExtensionDtype):
     @property
     def itemsize(self) -> int: ...
     @classmethod
-    def construct_array_type(cls) -> type[Self]: ...
+    def construct_array_type(cls) -> type[IntegerArray]: ...
     def __from_arrow__(self, array): ...
 
 class IntegerArray(BaseMaskedArray):
@@ -45,7 +44,7 @@ class IntegerArray(BaseMaskedArray):
         key: Sequence[int] | slice | npt.NDArray[np.integer],
         value: float | NAType | Sequence[float | NAType] | npt.NDArray[np.integer],
     ) -> None: ...
-    @overload  # type: ignore[override]
+    @overload
     def __getitem__(self, item: int) -> int | NAType: ...
     @overload
     def __getitem__(

pandas-stubs/core/arrays/interval.pyi

@@ -12,12 +12,13 @@ from pandas import (
     Index,
     Series,
 )
-from pandas.core.arrays.base import ExtensionArray as ExtensionArray
+from pandas.core.arrays.base import ExtensionArray
+from pandas.core.base import IndexOpsMixin
 from typing_extensions import Self
 
 from pandas._libs.interval import (
-    Interval as Interval,
-    IntervalMixin as IntervalMixin,
+    Interval,
+    IntervalMixin,
 )
 from pandas._typing import (
     Axis,
@@ -41,43 +42,90 @@ class IntervalArray(IntervalMixin, ExtensionArray, Generic[IntervalT]):
     @overload
     @classmethod
     def from_breaks(
-        cls, breaks: Sequence[int], closed: str = ..., copy: bool = ..., dtype=...
+        cls,
+        breaks: Sequence[int] | npt.NDArray[np.integer] | IndexOpsMixin[int],
+        closed: Literal["left", "right", "both", "neither"] = ...,
+        copy: bool = ...,
+        dtype: pd.IntervalDtype = ...,
     ) -> IntervalArray[Interval[int]]: ...
     @overload
     @classmethod
     def from_breaks(
-        cls, breaks: Sequence[float], closed: str = ..., copy: bool = ..., dtype=...
+        cls,
+        breaks: Sequence[float] | npt.NDArray[np.floating] | IndexOpsMixin[float],
+        closed: Literal["left", "right", "both", "neither"] = ...,
+        copy: bool = ...,
+        dtype: pd.IntervalDtype = ...,
     ) -> IntervalArray[Interval[float]]: ...
     @overload
     @classmethod
     def from_breaks(
         cls,
-        breaks: Sequence[pd.Timestamp | np.datetime64 | dt.datetime],
-        closed: str = ...,
+        breaks: Sequence[np.datetime64 | dt.datetime] | IndexOpsMixin[pd.Timestamp],
+        closed: Literal["left", "right", "both", "neither"] = ...,
         copy: bool = ...,
-        dtype=...,
+        dtype: pd.IntervalDtype = ...,
     ) -> IntervalArray[Interval[pd.Timestamp]]: ...
     @overload
     @classmethod
     def from_breaks(
         cls,
-        breaks: Sequence[pd.Timedelta | np.timedelta64 | dt.timedelta],
-        closed: str = ...,
+        breaks: Sequence[np.timedelta64 | dt.timedelta] | IndexOpsMixin[pd.Timedelta],
+        closed: Literal["left", "right", "both", "neither"] = ...,
         copy: bool = ...,
-        dtype=...,
+        dtype: pd.IntervalDtype = ...,
     ) -> IntervalArray[Interval[pd.Timedelta]]: ...
+    @overload
     @classmethod
     def from_arrays(
-        cls, left, right, closed: str = ..., copy: bool = ..., dtype=...
-    ) -> IntervalArray: ...
+        cls,
+        left: Sequence[int] | npt.NDArray[np.integer] | IndexOpsMixin[int],
+        right: Sequence[int] | npt.NDArray[np.integer] | IndexOpsMixin[int],
+        closed: Literal["left", "right", "both", "neither"] = ...,
+        copy: bool = ...,
+        dtype: pd.IntervalDtype = ...,
+    ) -> IntervalArray[Interval[int]]: ...
+    @overload
+    @classmethod
+    def from_arrays(
+        cls,
+        left: Sequence[float] | npt.NDArray[np.floating] | IndexOpsMixin[float],
+        right: Sequence[float] | npt.NDArray[np.floating] | IndexOpsMixin[float],
+        closed: Literal["left", "right", "both", "neither"] = ...,
+        copy: bool = ...,
+        dtype: pd.IntervalDtype = ...,
+    ) -> IntervalArray[Interval[float]]: ...
+    @overload
+    @classmethod
+    def from_arrays(
+        cls,
+        left: Sequence[np.datetime64 | dt.datetime] | pd.DatetimeIndex,
+        right: Sequence[np.datetime64 | dt.datetime] | pd.DatetimeIndex,
+        closed: Literal["left", "right", "both", "neither"] = ...,
+        copy: bool = ...,
+        dtype: pd.IntervalDtype = ...,
+    ) -> IntervalArray[Interval[pd.Timestamp]]: ...
+    @overload
+    @classmethod
+    def from_arrays(
+        cls,
+        left: Sequence[np.timedelta64 | dt.timedelta] | pd.TimedeltaIndex,
+        right: Sequence[np.timedelta64 | dt.timedelta] | pd.TimedeltaIndex,
+        closed: Literal["left", "right", "both", "neither"] = ...,
+        copy: bool = ...,
+        dtype: pd.IntervalDtype = ...,
+    ) -> IntervalArray[Interval[pd.Timedelta]]: ...
     @classmethod
     def from_tuples(
-        cls, data, closed: str = ..., copy: bool = ..., dtype=...
+        cls,
+        data,
+        closed: Literal["left", "right", "both", "neither"] = ...,
+        copy: bool = ...,
+        dtype: pd.IntervalDtype = ...,
     ) -> IntervalArray: ...
     def __iter__(self) -> IntervalT: ...
     def __len__(self) -> int: ...
     def __getitem__(self, value: IntervalT): ...
-    def __setitem__(self, key: int, value: IntervalT) -> None: ...
     def __eq__(self, other): ...
     def __ne__(self, other): ...
     def fillna(self, value=..., method=..., limit=...): ...

pandas-stubs/core/arrays/period.pyi

@@ -8,7 +8,6 @@ from pandas.core.arrays.datetimelike import (
 
 from pandas._libs.tslibs import Period
 from pandas._libs.tslibs.timestamps import Timestamp
-from pandas._typing import npt as npt
 
 from pandas.tseries.offsets import Tick
 

pandas-stubs/core/arrays/string_.pyi

@@ -48,7 +48,7 @@ class StringArray(PandasArray):
         | npt.NDArray[np.string_]
         | StringArray,
     ) -> None: ...
-    @overload  # type: ignore[override]
+    @overload
     def __getitem__(self, item: int) -> str | NAType: ...
     @overload
     def __getitem__(self, item: _ArrayKey) -> StringArray: ...

pandas-stubs/core/series.pyi

@@ -221,6 +221,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
         cls,
         data: DatetimeIndex | Sequence[Timestamp | np.datetime64 | datetime],
         index: Axes | None = ...,
+        *,
         dtype: TimestampDtypeArg = ...,
         name: Hashable | None = ...,
         copy: bool = ...,
@@ -240,6 +241,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
         cls,
         data: PeriodIndex,
         index: Axes | None = ...,
+        *,
         dtype: PeriodDtype = ...,
         name: Hashable | None = ...,
         copy: bool = ...,
@@ -249,6 +251,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
         cls,
         data: TimedeltaIndex | Sequence[Timedelta | np.timedelta64 | timedelta],
         index: Axes | None = ...,
+        *,
         dtype: TimedeltaDtypeArg = ...,
         name: Hashable | None = ...,
         copy: bool = ...,
@@ -260,6 +263,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
         | Interval[_OrderableT]
         | Sequence[Interval[_OrderableT]],
         index: Axes | None = ...,
+        *,
         dtype: Literal["Interval"] = ...,
         name: Hashable | None = ...,
         copy: bool = ...,
@@ -268,8 +272,9 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __new__(
         cls,
         data: object | _ListLike | Series[S1] | dict[int, S1] | dict[_str, S1] | None,
-        dtype: type[S1],
         index: Axes | None = ...,
+        *,
+        dtype: type[S1],
         name: Hashable | None = ...,
         copy: bool = ...,
     ) -> Self: ...
@@ -278,6 +283,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
         cls,
         data: Series[S1] | dict[int, S1] | dict[_str, S1] = ...,
         index: Axes | None = ...,
+        *,
         dtype: Dtype = ...,
         name: Hashable | None = ...,
         copy: bool = ...,
@@ -287,6 +293,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
         cls,
         data: object | _ListLike | None = ...,
         index: Axes | None = ...,
+        *,
         dtype: Dtype = ...,
         name: Hashable | None = ...,
         copy: bool = ...,

pyproject.toml

@@ -39,7 +39,7 @@ numpy = [
 ]
 
 [tool.poetry.dev-dependencies]
-mypy = "1.5.0"
+mypy = "1.5.1"
 pandas = "2.0.3"
 pyarrow = ">=10.0.1"
 pytest = ">=7.1.2"

tests/extension/decimal/array.py

@@ -3,6 +3,10 @@
 import decimal
 import numbers
 import sys
+from typing import (
+    Sequence,
+    cast,
+)
 
 import numpy as np
 import pandas as pd
@@ -200,13 +204,15 @@ def astype(self, dtype, copy=True):
 
         return super().astype(dtype, copy=copy)
 
-    def __setitem__(self, key, value) -> None:
+    def __setitem__(
+        self, key, value: DecimalArray | decimal.Decimal | Sequence[decimal.Decimal]
+    ) -> None:
         if is_list_like(value):
             if is_scalar(key):
                 raise ValueError("setting an array element with a sequence.")
-            value = [decimal.Decimal(v) for v in value]
+            value = [decimal.Decimal(v) for v in cast(Sequence[decimal.Decimal], value)]
         else:
-            value = decimal.Decimal(value)
+            value = decimal.Decimal(cast(decimal.Decimal, value))
 
         key = check_array_indexer(self, key)
         self._data[key] = value