MAINT: Bump pandas to 1.5.0

pandas-stubs/_config/config.pyi

@@ -47,7 +47,6 @@ class DisplayUnicode(DictWrapper):
 class Display(DictWrapper):
     chop_threshold: int | None
     colheader_justify: str
-    column_space: int
     date_dayfirst: bool
     date_yearfirst: bool
     encoding: str

pandas-stubs/core/algorithms.pyi

@@ -26,7 +26,8 @@ def unique(values: ExtensionArray) -> ExtensionArray: ...
 def factorize(
     values: Any,
     sort: bool = ...,
-    na_sentinel: int | None = ...,
+    # Not actually positional-only, used to handle deprecations in 1.5.0
+    *,
     use_na_sentinel: bool = ...,
     size_hint: int | None = ...,
 ) -> tuple[np.ndarray, np.ndarray | Index]: ...

pandas-stubs/core/arrays/base.pyi

@@ -33,7 +33,7 @@ class ExtensionArray:
     def astype(self, dtype, copy: bool = ...): ...
     def isna(self) -> ArrayLike: ...
     def argsort(
-        self, ascending: bool = ..., kind: str = ..., *args, **kwargs
+        self, *, ascending: bool = ..., kind: str = ..., **kwargs
     ) -> np.ndarray: ...
     def fillna(self, value=..., method=..., limit=...): ...
     def dropna(self): ...

pandas-stubs/core/arrays/categorical.pyi

@@ -42,8 +42,6 @@ class Categorical(ExtensionArray, PandasObject):
     ) -> None: ...
     @property
     def categories(self): ...
-    @categories.setter
-    def categories(self, categories) -> None: ...
     @property
     def ordered(self) -> Ordered: ...
     @property
@@ -64,22 +62,9 @@ class Categorical(ExtensionArray, PandasObject):
     ) -> Categorical: ...
     @property
     def codes(self) -> np_ndarray_int: ...
-    @overload
-    def set_ordered(self, value, inplace: Literal[True]) -> None: ...
-    @overload
-    def set_ordered(self, value, inplace: Literal[False] = ...) -> Categorical: ...
-    @overload
-    def set_ordered(self, value, inplace: bool = ...) -> Categorical | None: ...
-    @overload
-    def as_ordered(self, inplace: Literal[True]) -> None: ...
-    @overload
-    def as_ordered(self, inplace: Literal[False] = ...) -> Categorical: ...
-    @overload
-    def as_ordered(self, inplace: bool = ...) -> Categorical | None: ...
-    @overload
-    def as_unordered(self, inplace: Literal[True]) -> None: ...
-    @overload
-    def as_unordered(self, inplace: Literal[False] = ...) -> Categorical: ...
+    def set_ordered(self, value) -> Categorical: ...
+    def as_ordered(self) -> Categorical: ...
+    def as_unordered(self) -> Categorical: ...
     @overload
     def set_categories(
         self, new_categories, ordered=..., rename: bool = ..., *, inplace: Literal[True]

pandas-stubs/core/base.pyi

@@ -17,10 +17,6 @@ from pandas._typing import (
     npt,
 )
 
-# TODO: These two moved to pandas.errors after switch to 1.5.x
-class DataError(Exception): ...
-class SpecificationError(Exception): ...
-
 class PandasObject(DirNamesMixin):
     def __sizeof__(self) -> int: ...
 
@@ -80,13 +76,11 @@ class IndexOpsMixin:
     @property
     def is_unique(self) -> bool: ...
     @property
-    def is_monotonic(self) -> bool: ...
-    @property
     def is_monotonic_decreasing(self) -> bool: ...
     @property
     def is_monotonic_increasing(self) -> bool: ...
     def factorize(
-        self, sort: bool = ..., na_sentinel: int = ...
+        self, sort: bool = ...
     ) -> tuple[np.ndarray, np.ndarray | Index | Categorical]: ...
     def searchsorted(
         self, value, side: Literal["left", "right"] = ..., sorter=...

pandas-stubs/core/common.pyi

@@ -5,10 +5,6 @@ from typing import (
 
 from pandas._typing import T
 
-# TODO: Remove from here after 1.5.x is target, moved to pandas.errors
-class SettingWithCopyError(ValueError): ...
-class SettingWithCopyWarning(Warning): ...
-
 def flatten(line) -> None: ...
 def consensus_name_attr(objs): ...
 def is_bool_indexer(key) -> bool: ...

pandas-stubs/core/computation/engines.pyi

@@ -1,8 +1,5 @@
 import abc
 
-# TODO: Remove from here after 1.5.x, moved to pandas.errors
-class NumExprClobberingError(NameError): ...
-
 class AbstractEngine(metaclass=abc.ABCMeta):
     has_neg_frac: bool = ...
     expr = ...

pandas-stubs/core/frame.pyi

@@ -22,10 +22,7 @@ from pandas import (
 )
 from pandas.core.arraylike import OpsMixin
 from pandas.core.generic import NDFrame
-from pandas.core.groupby.generic import (
-    _DataFrameGroupByNonScalar,
-    _DataFrameGroupByScalar,
-)
+from pandas.core.groupby.generic import DataFrameGroupBy
 from pandas.core.groupby.grouper import Grouper
 from pandas.core.indexes.base import Index
 from pandas.core.indexing import (
@@ -208,7 +205,6 @@ class DataFrame(NDFrame, OpsMixin):
     @property
     def style(self) -> Styler: ...
     def items(self) -> Iterable[tuple[Hashable, Series]]: ...
-    def iteritems(self) -> Iterable[tuple[Label, Series]]: ...
     def iterrows(self) -> Iterable[tuple[Label, Series]]: ...
     def itertuples(self, index: _bool = ..., name: _str | None = ...): ...
     def __len__(self) -> int: ...
@@ -294,6 +290,7 @@ class DataFrame(NDFrame, OpsMixin):
     def to_stata(
         self,
         path: FilePath | WriteBuffer[bytes],
+        *,
         convert_dates: dict[HashableT, StataDateFormat] | None = ...,
         write_index: _bool = ...,
         byteorder: Literal["<", ">", "little", "big"] | None = ...,
@@ -304,7 +301,6 @@ class DataFrame(NDFrame, OpsMixin):
         convert_strl: list[HashableT] | None = ...,
         compression: CompressionOptions = ...,
         storage_options: StorageOptions = ...,
-        *,
         value_labels: dict[Hashable, dict[float, str]] | None = ...,
     ) -> None: ...
     def to_feather(self, path: FilePath | WriteBuffer[bytes], **kwargs) -> None: ...
@@ -992,23 +988,9 @@ class DataFrame(NDFrame, OpsMixin):
         filter_func: Callable | None = ...,
         errors: IgnoreRaise = ...,
     ) -> None: ...
-    @overload
-    def groupby(
-        self,
-        by: Scalar,
-        axis: AxisType = ...,
-        level: Level | None = ...,
-        as_index: _bool = ...,
-        sort: _bool = ...,
-        group_keys: _bool = ...,
-        squeeze: _bool = ...,
-        observed: _bool = ...,
-        dropna: _bool = ...,
-    ) -> _DataFrameGroupByScalar: ...
-    @overload
     def groupby(
         self,
-        by: GroupByObjectNonScalar | None = ...,
+        by: Scalar | GroupByObjectNonScalar | None = ...,
         axis: AxisType = ...,
         level: Level | None = ...,
         as_index: _bool = ...,
@@ -1017,9 +999,10 @@ class DataFrame(NDFrame, OpsMixin):
         squeeze: _bool = ...,
         observed: _bool = ...,
         dropna: _bool = ...,
-    ) -> _DataFrameGroupByNonScalar: ...
+    ) -> DataFrameGroupBy: ...
     def pivot(
         self,
+        *,
         index: IndexLabel = ...,
         columns: IndexLabel = ...,
         values: IndexLabel = ...,
@@ -1126,14 +1109,18 @@ class DataFrame(NDFrame, OpsMixin):
         self,
         method: Literal["pearson", "kendall", "spearman"] = ...,
         min_periods: int = ...,
+        numeric_only: _bool = ...,
+    ) -> DataFrame: ...
+    def cov(
+        self, min_periods: int | None = ..., ddof: int = ..., numeric_only: _bool = ...
     ) -> DataFrame: ...
-    def cov(self, min_periods: int | None = ..., ddof: int = ...) -> DataFrame: ...
     def corrwith(
         self,
         other: DataFrame | Series,
         axis: AxisType | None = ...,
         drop: _bool = ...,
         method: Literal["pearson", "kendall", "spearman"] = ...,
+        numeric_only: _bool = ...,
     ) -> Series: ...
     @overload
     def count(
@@ -1144,8 +1131,12 @@ class DataFrame(NDFrame, OpsMixin):
         self, axis: AxisType = ..., level: None = ..., numeric_only: _bool = ...
     ) -> Series: ...
     def nunique(self, axis: AxisType = ..., dropna: bool = ...) -> Series: ...
-    def idxmax(self, axis: AxisType = ..., skipna: _bool = ...) -> Series: ...
-    def idxmin(self, axis: AxisType = ..., skipna: _bool = ...) -> Series: ...
+    def idxmax(
+        self, axis: AxisType = ..., skipna: _bool = ..., numeric_only: _bool = ...
+    ) -> Series: ...
+    def idxmin(
+        self, axis: AxisType = ..., skipna: _bool = ..., numeric_only: _bool = ...
+    ) -> Series: ...
     @overload
     def mode(
         self,
@@ -1298,6 +1289,7 @@ class DataFrame(NDFrame, OpsMixin):
     @overload
     def any(
         self,
+        *,
         axis: None,
         bool_only: _bool | None = ...,
         skipna: _bool = ...,
@@ -1306,6 +1298,7 @@ class DataFrame(NDFrame, OpsMixin):
     @overload
     def any(
         self,
+        *,
         axis: AxisType = ...,
         bool_only: _bool | None = ...,
         skipna: _bool = ...,
@@ -1523,15 +1516,15 @@ class DataFrame(NDFrame, OpsMixin):
         axis: AxisType | None = ...,
         skipna: _bool | None = ...,
         level: None = ...,
-        numeric_only: _bool | None = ...,
+        numeric_only: _bool = ...,
         **kwargs,
     ) -> Series: ...
     def kurtosis(
         self,
         axis: AxisType | None = ...,
         skipna: _bool | None = ...,
         level: None = ...,
-        numeric_only: _bool | None = ...,
+        numeric_only: _bool = ...,
         **kwargs,
     ) -> Series: ...
     def last(self, offset) -> DataFrame: ...
@@ -1542,62 +1535,46 @@ class DataFrame(NDFrame, OpsMixin):
     def lt(
         self, other, axis: AxisType = ..., level: Level | None = ...
     ) -> DataFrame: ...
-    @overload
-    def mad(
-        self,
-        axis: AxisType | None = ...,
-        skipna: _bool | None = ...,
-        level: None = ...,
-    ) -> Series: ...
-    @overload
-    def mad(
-        self,
-        axis: AxisType | None = ...,
-        skipna: _bool | None = ...,
-        *,
-        level: Level,
-        **kwargs,
-    ) -> DataFrame: ...
     def mask(
         self,
         cond: Series | DataFrame | np.ndarray,
         other=...,
         inplace: _bool = ...,
         axis: AxisType | None = ...,
         level: Level | None = ...,
-        errors: _str = ...,
+        *,  # Not actually positional-only, but needed due to depr in 1.5.0
         try_cast: _bool = ...,
     ) -> DataFrame: ...
     def max(
         self,
         axis: AxisType | None = ...,
         skipna: _bool | None = ...,
         level: None = ...,
-        numeric_only: _bool | None = ...,
+        numeric_only: _bool = ...,
         **kwargs,
     ) -> Series: ...
     def mean(
         self,
         axis: AxisType | None = ...,
         skipna: _bool | None = ...,
         level: None = ...,
-        numeric_only: _bool | None = ...,
+        numeric_only: _bool = ...,
         **kwargs,
     ) -> Series: ...
     def median(
         self,
         axis: AxisType | None = ...,
         skipna: _bool | None = ...,
         level: None = ...,
-        numeric_only: _bool | None = ...,
+        numeric_only: _bool = ...,
         **kwargs,
     ) -> Series: ...
     def min(
         self,
         axis: AxisType | None = ...,
         skipna: _bool | None = ...,
         level: None = ...,
-        numeric_only: _bool | None = ...,
+        numeric_only: _bool = ...,
         **kwargs,
     ) -> Series: ...
     def mod(
@@ -1651,7 +1628,7 @@ class DataFrame(NDFrame, OpsMixin):
         axis: AxisType | None = ...,
         skipna: _bool | None = ...,
         level: None = ...,
-        numeric_only: _bool | None = ...,
+        numeric_only: _bool = ...,
         min_count: int = ...,
         **kwargs,
     ) -> Series: ...
@@ -1660,7 +1637,7 @@ class DataFrame(NDFrame, OpsMixin):
         axis: AxisType | None = ...,
         skipna: _bool = ...,
         level: None = ...,
-        numeric_only: _bool | None = ...,
+        numeric_only: _bool = ...,
         min_count: int = ...,
         **kwargs,
     ) -> Series: ...
@@ -1830,32 +1807,17 @@ class DataFrame(NDFrame, OpsMixin):
         skipna: _bool | None = ...,
         level: None = ...,
         ddof: int = ...,
-        numeric_only: _bool | None = ...,
+        numeric_only: _bool = ...,
         **kwargs,
     ) -> Series: ...
-    @overload
-    def set_axis(
-        self, labels, inplace: Literal[True], axis: AxisType = ...
-    ) -> None: ...
-    @overload
-    def set_axis(
-        self, labels, inplace: Literal[False], axis: AxisType = ...
-    ) -> DataFrame: ...
-    @overload
-    def set_axis(self, labels, *, axis: AxisType = ...) -> DataFrame: ...
-    @overload
-    def set_axis(
-        self,
-        labels,
-        axis: AxisType = ...,
-        inplace: _bool | None = ...,
-    ) -> DataFrame | None: ...
+    # Not actually positional, but used to handle removal of deprecated
+    def set_axis(self, labels, *, axis: AxisType, copy: _bool = ...) -> DataFrame: ...
     def skew(
         self,
         axis: AxisType | None = ...,
         skipna: _bool | None = ...,
         level: None = ...,
-        numeric_only: _bool | None = ...,
+        numeric_only: _bool = ...,
         **kwargs,
     ) -> Series: ...
     def slice_shift(self, periods: int = ..., axis: AxisType = ...) -> DataFrame: ...
@@ -1888,7 +1850,7 @@ class DataFrame(NDFrame, OpsMixin):
         axis: AxisType | None = ...,
         skipna: _bool | None = ...,
         level: None = ...,
-        numeric_only: _bool | None = ...,
+        numeric_only: _bool = ...,
         min_count: int = ...,
         **kwargs,
     ) -> Series: ...
@@ -2024,7 +1986,7 @@ class DataFrame(NDFrame, OpsMixin):
         skipna: _bool | None = ...,
         level: None = ...,
         ddof: int = ...,
-        numeric_only: _bool | None = ...,
+        numeric_only: _bool = ...,
         **kwargs,
     ) -> Series: ...
     def where(
@@ -2038,7 +2000,7 @@ class DataFrame(NDFrame, OpsMixin):
         inplace: _bool = ...,
         axis: AxisType | None = ...,
         level: Level | None = ...,
-        errors: _str = ...,
+        *,  # Not actually positional-only, but needed due to depr in 1.5.0
         try_cast: _bool = ...,
     ) -> DataFrame: ...
     # Move from generic because Series is Generic and it returns Series[bool] there