Skip to content

Commit 03ffe84

Browse files
committed
type return as Series[Any]
1 parent eb6c70b commit 03ffe84

File tree

2 files changed

+5
-8
lines changed

2 files changed

+5
-8
lines changed

pandas-stubs/core/series.pyi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1223,7 +1223,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
12231223
freq: Frequency | timedelta | None = ...,
12241224
axis: AxisIndex = ...,
12251225
fill_value: object | None = ...,
1226-
) -> Series[S1]: ...
1226+
) -> Series[Any]: ...
12271227
def memory_usage(self, index: _bool = ..., deep: _bool = ...) -> int: ...
12281228
def isin(self, values: Iterable | Series[S1] | dict) -> Series[_bool]: ...
12291229
def between(

tests/test_series.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -417,19 +417,16 @@ def test_types_sort_values_with_key() -> None:
417417

418418
def test_types_shift() -> None:
419419
s = pd.Series([1, 2, 3], index=pd.date_range("2020", periods=3))
420-
# Return type "Series[int]"" not quite correct
421-
# - https://github.com/pandas-dev/pandas-stubs/issues/1111
422-
# - https://github.com/pandas-dev/pandas-stubs/issues/1110
423-
check(assert_type(s.shift(), "pd.Series[int]"), pd.Series, np.floating)
420+
check(assert_type(s.shift(), "pd.Series[Any]"), pd.Series, np.floating)
424421
check(
425-
assert_type(s.shift(axis=0, periods=1), "pd.Series[int]"),
422+
assert_type(s.shift(axis=0, periods=1), "pd.Series[Any]"),
426423
pd.Series,
427424
np.floating,
428425
)
429426
check(
430-
assert_type(s.shift(-1, fill_value=0), "pd.Series[int]"), pd.Series, np.integer
427+
assert_type(s.shift(-1, fill_value=0), "pd.Series[Any]"), pd.Series, np.integer
431428
)
432-
check(assert_type(s.shift(freq="1D"), "pd.Series[int]"), pd.Series, np.integer)
429+
check(assert_type(s.shift(freq="1D"), "pd.Series[Any]"), pd.Series, np.integer)
433430

434431

435432
def test_types_rank() -> None:

0 commit comments

Comments
 (0)