@@ -17,7 +17,6 @@ from pandas import (
1717 Series ,
1818 TimedeltaIndex ,
1919)
20- from pandas .core .series import TimedeltaSeries
2120from typing_extensions import (
2221 Self ,
2322 TypeAlias ,
@@ -162,10 +161,7 @@ class Timedelta(timedelta):
162161 @overload
163162 def __add__ (self , other : pd .TimedeltaIndex ) -> pd .TimedeltaIndex : ...
164163 @overload
165- def __add__ (
166- self ,
167- other : TimedeltaSeries ,
168- ) -> TimedeltaSeries : ...
164+ def __add__ (self , other : Series [Timedelta ]) -> Series [Timedelta ]: ...
169165 @overload
170166 def __add__ (self , other : Series [Timestamp ]) -> Series [Timestamp ]: ...
171167 @overload
@@ -198,9 +194,7 @@ class Timedelta(timedelta):
198194 @overload
199195 def __sub__ (self , other : pd .TimedeltaIndex ) -> TimedeltaIndex : ...
200196 @overload
201- def __sub__ (
202- self , other : TimedeltaSeries | Series [pd .Timedelta ]
203- ) -> TimedeltaSeries : ...
197+ def __sub__ (self , other : Series [pd .Timedelta ]) -> Series [pd .Timedelta ]: ...
204198 @overload
205199 def __rsub__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> Timedelta : ...
206200 @overload
@@ -234,9 +228,9 @@ class Timedelta(timedelta):
234228 self , other : np_ndarray [ShapeT , np .integer ] | np_ndarray [ShapeT , np .floating ]
235229 ) -> np_ndarray [ShapeT , np .timedelta64 ]: ...
236230 @overload
237- def __mul__ (self , other : Series [int ]) -> TimedeltaSeries : ...
231+ def __mul__ (self , other : Series [int ]) -> Series [ Timedelta ] : ...
238232 @overload
239- def __mul__ (self , other : Series [float ]) -> TimedeltaSeries : ...
233+ def __mul__ (self , other : Series [float ]) -> Series [ Timedelta ] : ...
240234 @overload
241235 def __mul__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
242236 @overload
@@ -246,9 +240,9 @@ class Timedelta(timedelta):
246240 self , other : np_ndarray [ShapeT , np .floating ] | np_ndarray [ShapeT , np .integer ]
247241 ) -> np_ndarray [ShapeT , np .timedelta64 ]: ...
248242 @overload
249- def __rmul__ (self , other : Series [int ]) -> TimedeltaSeries : ...
243+ def __rmul__ (self , other : Series [int ]) -> Series [ Timedelta ] : ...
250244 @overload
251- def __rmul__ (self , other : Series [float ]) -> TimedeltaSeries : ...
245+ def __rmul__ (self , other : Series [float ]) -> Series [ Timedelta ] : ...
252246 # maybe related to https://github.com/python/mypy/issues/10755
253247 @overload
254248 def __rmul__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
@@ -269,11 +263,11 @@ class Timedelta(timedelta):
269263 @overload
270264 def __floordiv__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
271265 @overload
272- def __floordiv__ (self , other : Series [int ]) -> TimedeltaSeries : ...
266+ def __floordiv__ (self , other : Series [int ]) -> Series [ Timedelta ] : ...
273267 @overload
274- def __floordiv__ (self , other : Series [float ]) -> TimedeltaSeries : ...
268+ def __floordiv__ (self , other : Series [float ]) -> Series [ Timedelta ] : ...
275269 @overload
276- def __floordiv__ (self , other : TimedeltaSeries ) -> Series [int ]: ...
270+ def __floordiv__ (self , other : Series [ Timedelta ] ) -> Series [int ]: ...
277271 @overload
278272 def __floordiv__ (self , other : NaTType | None ) -> float : ...
279273 @overload
@@ -294,19 +288,19 @@ class Timedelta(timedelta):
294288 self , other : np_ndarray [ShapeT , np .integer ] | np_ndarray [ShapeT , np .floating ]
295289 ) -> np_ndarray [ShapeT , np .timedelta64 ]: ...
296290 @overload
297- def __truediv__ (self , other : TimedeltaSeries ) -> Series [float ]: ...
291+ def __truediv__ (self , other : Series [ Timedelta ] ) -> Series [float ]: ...
298292 @overload
299- def __truediv__ (self , other : Series [int ]) -> TimedeltaSeries : ...
293+ def __truediv__ (self , other : Series [int ]) -> Series [ Timedelta ] : ...
300294 @overload
301- def __truediv__ (self , other : Series [float ]) -> TimedeltaSeries : ...
295+ def __truediv__ (self , other : Series [float ]) -> Series [ Timedelta ] : ...
302296 @overload
303297 def __truediv__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
304298 def __rtruediv__ (self , other : timedelta | Timedelta | NaTType ) -> float : ...
305299 # Override due to more types supported than dt.timedelta
306300 @overload
307301 def __eq__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
308302 @overload
309- def __eq__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ... # type: ignore[overload-overlap]
303+ def __eq__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ... # type: ignore[overload-overlap]
310304 @overload
311305 def __eq__ (self , other : Index ) -> np_1darray [np .bool ]: ... # type: ignore[overload-overlap]
312306 @overload
@@ -319,7 +313,7 @@ class Timedelta(timedelta):
319313 @overload
320314 def __ne__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
321315 @overload
322- def __ne__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ... # type: ignore[overload-overlap]
316+ def __ne__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ... # type: ignore[overload-overlap]
323317 @overload
324318 def __ne__ (self , other : Index ) -> np_1darray [np .bool ]: ... # type: ignore[overload-overlap]
325319 @overload
@@ -334,7 +328,7 @@ class Timedelta(timedelta):
334328 @overload
335329 def __mod__ (self , other : float ) -> Timedelta : ...
336330 @overload
337- def __mod__ (self , other : Series [int ] | Series [float ]) -> TimedeltaSeries : ...
331+ def __mod__ (self , other : Series [int ] | Series [float ]) -> Series [ Timedelta ] : ...
338332 @overload
339333 def __mod__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
340334 @overload
@@ -343,8 +337,8 @@ class Timedelta(timedelta):
343337 ) -> np_ndarray [ShapeT , np .timedelta64 ]: ...
344338 @overload
345339 def __mod__ (
346- self , other : Series [int ] | Series [float ] | TimedeltaSeries
347- ) -> TimedeltaSeries : ...
340+ self , other : Series [int ] | Series [float ] | Series [ Timedelta ]
341+ ) -> Series [ Timedelta ] : ...
348342 def __divmod__ (self , other : timedelta ) -> tuple [int , Timedelta ]: ...
349343 # Mypy complains Forward operator "<inequality op>" is not callable, so ignore misc
350344 # for le, lt ge and gt
@@ -358,7 +352,7 @@ class Timedelta(timedelta):
358352 self , other : np_ndarray [ShapeT , np .timedelta64 ]
359353 ) -> np_ndarray [ShapeT , np .bool_ ]: ...
360354 @overload
361- def __le__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ...
355+ def __le__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ...
362356 # Override due to more types supported than dt.timedelta
363357 @overload # type: ignore[override]
364358 def __lt__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[misc]
@@ -369,7 +363,7 @@ class Timedelta(timedelta):
369363 self , other : np_ndarray [ShapeT , np .timedelta64 ]
370364 ) -> np_ndarray [ShapeT , np .bool_ ]: ...
371365 @overload
372- def __lt__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ...
366+ def __lt__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ...
373367 # Override due to more types supported than dt.timedelta
374368 @overload # type: ignore[override]
375369 def __ge__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[misc]
@@ -380,7 +374,7 @@ class Timedelta(timedelta):
380374 self , other : np_ndarray [ShapeT , np .timedelta64 ]
381375 ) -> np_ndarray [ShapeT , np .bool_ ]: ...
382376 @overload
383- def __ge__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ...
377+ def __ge__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ...
384378 # Override due to more types supported than dt.timedelta
385379 @overload # type: ignore[override]
386380 def __gt__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[misc]
@@ -391,7 +385,7 @@ class Timedelta(timedelta):
391385 self , other : np_ndarray [ShapeT , np .timedelta64 ]
392386 ) -> np_ndarray [ShapeT , np .bool_ ]: ...
393387 @overload
394- def __gt__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ...
388+ def __gt__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ...
395389 def __hash__ (self ) -> int : ...
396390 def isoformat (self ) -> str : ...
397391 def to_numpy (self ) -> np .timedelta64 : ...
0 commit comments