Skip to content

Commit c4482de

Browse files
committed
fix: typing
1 parent 61de1a1 commit c4482de

File tree

1 file changed

+32
-7
lines changed

1 file changed

+32
-7
lines changed

pandas-stubs/core/series.pyi

Lines changed: 32 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,7 @@ from pandas._typing import (
188188
np_ndarray_complex,
189189
np_ndarray_dt,
190190
np_ndarray_float,
191+
np_ndarray_td,
191192
npt,
192193
num,
193194
)
@@ -2146,12 +2147,24 @@ class Series(IndexOpsMixin[S1], NDFrame):
21462147
@overload
21472148
def __sub__(
21482149
self: Series[Timestamp],
2149-
other: timedelta | np.timedelta64 | TimedeltaSeries | TimedeltaIndex,
2150+
other: (
2151+
timedelta
2152+
| np.timedelta64
2153+
| np_ndarray_td
2154+
| TimedeltaSeries
2155+
| TimedeltaIndex
2156+
),
21502157
) -> TimestampSeries: ...
21512158
@overload
21522159
def __sub__(
21532160
self: Series[Timedelta],
2154-
other: timedelta | np.timedelta64 | TimedeltaSeries | TimedeltaIndex,
2161+
other: (
2162+
timedelta
2163+
| np.timedelta64
2164+
| np_ndarray_td
2165+
| TimedeltaSeries
2166+
| TimedeltaIndex
2167+
),
21552168
) -> TimedeltaSeries: ...
21562169
@overload
21572170
def sub(
@@ -2170,7 +2183,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
21702183
axis: int = 0,
21712184
) -> Series: ...
21722185
@overload
2173-
def sub(
2186+
def sub( # type: ignore[overload-overlap]
21742187
self,
21752188
other: Series[Never],
21762189
level: Level | None = None,
@@ -2264,19 +2277,31 @@ class Series(IndexOpsMixin[S1], NDFrame):
22642277
@overload
22652278
def sub(
22662279
self: Series[Timestamp],
2267-
other: timedelta | np.timedelta64 | TimedeltaSeries | TimedeltaIndex,
2280+
other: (
2281+
timedelta
2282+
| np.timedelta64
2283+
| np_ndarray_td
2284+
| TimedeltaSeries
2285+
| TimedeltaIndex
2286+
),
22682287
level: Level | None = None,
22692288
fill_value: float | None = None,
22702289
axis: int = 0,
2271-
) -> Series[complex]: ...
2290+
) -> TimestampSeries: ...
22722291
@overload
22732292
def sub(
22742293
self: Series[Timedelta],
2275-
other: timedelta | np.timedelta64 | TimedeltaSeries | TimedeltaIndex,
2294+
other: (
2295+
timedelta
2296+
| np.timedelta64
2297+
| np_ndarray_td
2298+
| TimedeltaSeries
2299+
| TimedeltaIndex
2300+
),
22762301
level: Level | None = None,
22772302
fill_value: float | None = None,
22782303
axis: int = 0,
2279-
) -> Series[complex]: ...
2304+
) -> TimedeltaSeries: ...
22802305
@overload
22812306
def __rsub__( # type: ignore[misc]
22822307
self: Series[Never],

0 commit comments

Comments
 (0)