From d5dd026dcf675c89e2f1c1ad64a23f7e2e9837e4 Mon Sep 17 00:00:00 2001 From: cmp0xff Date: Mon, 29 Sep 2025 17:29:00 +0200 Subject: [PATCH] fix(series): #1388 to_numpy --- pandas-stubs/core/series.pyi | 12 ++++++++++-- tests/series/test_series.py | 2 ++ 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/pandas-stubs/core/series.pyi b/pandas-stubs/core/series.pyi index 2100b4382..5ebd3d407 100644 --- a/pandas-stubs/core/series.pyi +++ b/pandas-stubs/core/series.pyi @@ -4613,9 +4613,17 @@ class Series(IndexOpsMixin[S1], NDFrame): ) -> _T: ... def to_list(self) -> list[S1]: ... @overload # type: ignore[override] + def to_numpy( + self: Series[Never], + dtype: DTypeLike | None = None, + copy: bool = False, + na_value: Scalar = ..., + **kwargs, + ) -> np_1darray: ... + @overload def to_numpy( # pyrefly: ignore[bad-override] self: Series[Timestamp], - dtype: None | type[np.datetime64] = None, + dtype: type[np.datetime64] | None = None, copy: bool = False, na_value: Scalar = ..., **kwargs, @@ -4631,7 +4639,7 @@ class Series(IndexOpsMixin[S1], NDFrame): @overload def to_numpy( # pyrefly: ignore[bad-override] self: Series[Timedelta], - dtype: None | type[np.timedelta64] = None, + dtype: type[np.timedelta64] | None = None, copy: bool = False, na_value: Scalar = ..., **kwargs, diff --git a/tests/series/test_series.py b/tests/series/test_series.py index 8160e0baa..bfa0c16ce 100644 --- a/tests/series/test_series.py +++ b/tests/series/test_series.py @@ -1981,6 +1981,8 @@ def test_types_to_numpy() -> None: check(assert_type(s.to_numpy(na_value=np.float16(4)), np_1darray), np_1darray) check(assert_type(s.to_numpy(na_value=np.complex128(4, 7)), np_1darray), np_1darray) + check(assert_type(pd.Series().to_numpy(), np_1darray), np_1darray) + def test_where() -> None: s = pd.Series([1, 2, 3], dtype=int)