diff --git a/pandas-stubs/core/series.pyi b/pandas-stubs/core/series.pyi index 91923f9e..21ea1d1b 100644 --- a/pandas-stubs/core/series.pyi +++ b/pandas-stubs/core/series.pyi @@ -74,6 +74,7 @@ from pandas._typing import ( AggFuncTypeBase, AggFuncTypeDictFrame, AggFuncTypeSeriesToFrame, + AnyArrayLike, ArrayLike, Axes, Axis, @@ -1065,8 +1066,8 @@ class Series(IndexOpsMixin, NDFrame, Generic[S1]): ) -> Scalar | Series[S1]: ... def clip( self, - lower: float | None = ..., - upper: float | None = ..., + lower: AnyArrayLike | float | None = ..., + upper: AnyArrayLike | float | None = ..., axis: SeriesAxisType | None = ..., inplace: _bool = ..., *args, diff --git a/tests/test_series.py b/tests/test_series.py index f37631d6..f17b7c7e 100644 --- a/tests/test_series.py +++ b/tests/test_series.py @@ -1330,3 +1330,11 @@ def test_logical_operators() -> None: pd.Series, bool, ) + + +def test_AnyArrayLike_and_clip() -> None: + ser = pd.Series([1, 2, 3]) + s1 = ser.clip(lower=ser) + s2 = ser.clip(upper=ser) + check(assert_type(s1, pd.Series), pd.Series) + check(assert_type(s2, pd.Series), pd.Series)