@@ -17,7 +17,6 @@ from pandas import (
17
17
Series ,
18
18
TimedeltaIndex ,
19
19
)
20
- from pandas .core .series import TimedeltaSeries
21
20
from typing_extensions import (
22
21
Self ,
23
22
TypeAlias ,
@@ -162,10 +161,7 @@ class Timedelta(timedelta):
162
161
@overload
163
162
def __add__ (self , other : pd .TimedeltaIndex ) -> pd .TimedeltaIndex : ...
164
163
@overload
165
- def __add__ (
166
- self ,
167
- other : TimedeltaSeries ,
168
- ) -> TimedeltaSeries : ...
164
+ def __add__ (self , other : Series [Timedelta ]) -> Series [Timedelta ]: ...
169
165
@overload
170
166
def __add__ (self , other : Series [Timestamp ]) -> Series [Timestamp ]: ...
171
167
@overload
@@ -198,9 +194,7 @@ class Timedelta(timedelta):
198
194
@overload
199
195
def __sub__ (self , other : pd .TimedeltaIndex ) -> TimedeltaIndex : ...
200
196
@overload
201
- def __sub__ (
202
- self , other : TimedeltaSeries | Series [pd .Timedelta ]
203
- ) -> TimedeltaSeries : ...
197
+ def __sub__ (self , other : Series [pd .Timedelta ]) -> Series [pd .Timedelta ]: ...
204
198
@overload
205
199
def __rsub__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> Timedelta : ...
206
200
@overload
@@ -234,9 +228,9 @@ class Timedelta(timedelta):
234
228
self , other : np_ndarray [ShapeT , np .integer ] | np_ndarray [ShapeT , np .floating ]
235
229
) -> np_ndarray [ShapeT , np .timedelta64 ]: ...
236
230
@overload
237
- def __mul__ (self , other : Series [int ]) -> TimedeltaSeries : ...
231
+ def __mul__ (self , other : Series [int ]) -> Series [ Timedelta ] : ...
238
232
@overload
239
- def __mul__ (self , other : Series [float ]) -> TimedeltaSeries : ...
233
+ def __mul__ (self , other : Series [float ]) -> Series [ Timedelta ] : ...
240
234
@overload
241
235
def __mul__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
242
236
@overload
@@ -246,9 +240,9 @@ class Timedelta(timedelta):
246
240
self , other : np_ndarray [ShapeT , np .floating ] | np_ndarray [ShapeT , np .integer ]
247
241
) -> np_ndarray [ShapeT , np .timedelta64 ]: ...
248
242
@overload
249
- def __rmul__ (self , other : Series [int ]) -> TimedeltaSeries : ...
243
+ def __rmul__ (self , other : Series [int ]) -> Series [ Timedelta ] : ...
250
244
@overload
251
- def __rmul__ (self , other : Series [float ]) -> TimedeltaSeries : ...
245
+ def __rmul__ (self , other : Series [float ]) -> Series [ Timedelta ] : ...
252
246
# maybe related to https://github.com/python/mypy/issues/10755
253
247
@overload
254
248
def __rmul__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
@@ -269,11 +263,11 @@ class Timedelta(timedelta):
269
263
@overload
270
264
def __floordiv__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
271
265
@overload
272
- def __floordiv__ (self , other : Series [int ]) -> TimedeltaSeries : ...
266
+ def __floordiv__ (self , other : Series [int ]) -> Series [ Timedelta ] : ...
273
267
@overload
274
- def __floordiv__ (self , other : Series [float ]) -> TimedeltaSeries : ...
268
+ def __floordiv__ (self , other : Series [float ]) -> Series [ Timedelta ] : ...
275
269
@overload
276
- def __floordiv__ (self , other : TimedeltaSeries ) -> Series [int ]: ...
270
+ def __floordiv__ (self , other : Series [ Timedelta ] ) -> Series [int ]: ...
277
271
@overload
278
272
def __floordiv__ (self , other : NaTType | None ) -> float : ...
279
273
@overload
@@ -294,19 +288,19 @@ class Timedelta(timedelta):
294
288
self , other : np_ndarray [ShapeT , np .integer ] | np_ndarray [ShapeT , np .floating ]
295
289
) -> np_ndarray [ShapeT , np .timedelta64 ]: ...
296
290
@overload
297
- def __truediv__ (self , other : TimedeltaSeries ) -> Series [float ]: ...
291
+ def __truediv__ (self , other : Series [ Timedelta ] ) -> Series [float ]: ...
298
292
@overload
299
- def __truediv__ (self , other : Series [int ]) -> TimedeltaSeries : ...
293
+ def __truediv__ (self , other : Series [int ]) -> Series [ Timedelta ] : ...
300
294
@overload
301
- def __truediv__ (self , other : Series [float ]) -> TimedeltaSeries : ...
295
+ def __truediv__ (self , other : Series [float ]) -> Series [ Timedelta ] : ...
302
296
@overload
303
297
def __truediv__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
304
298
def __rtruediv__ (self , other : timedelta | Timedelta | NaTType ) -> float : ...
305
299
# Override due to more types supported than dt.timedelta
306
300
@overload
307
301
def __eq__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
308
302
@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]
310
304
@overload
311
305
def __eq__ (self , other : Index ) -> np_1darray [np .bool ]: ... # type: ignore[overload-overlap]
312
306
@overload
@@ -319,7 +313,7 @@ class Timedelta(timedelta):
319
313
@overload
320
314
def __ne__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
321
315
@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]
323
317
@overload
324
318
def __ne__ (self , other : Index ) -> np_1darray [np .bool ]: ... # type: ignore[overload-overlap]
325
319
@overload
@@ -334,7 +328,7 @@ class Timedelta(timedelta):
334
328
@overload
335
329
def __mod__ (self , other : float ) -> Timedelta : ...
336
330
@overload
337
- def __mod__ (self , other : Series [int ] | Series [float ]) -> TimedeltaSeries : ...
331
+ def __mod__ (self , other : Series [int ] | Series [float ]) -> Series [ Timedelta ] : ...
338
332
@overload
339
333
def __mod__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
340
334
@overload
@@ -343,8 +337,8 @@ class Timedelta(timedelta):
343
337
) -> np_ndarray [ShapeT , np .timedelta64 ]: ...
344
338
@overload
345
339
def __mod__ (
346
- self , other : Series [int ] | Series [float ] | TimedeltaSeries
347
- ) -> TimedeltaSeries : ...
340
+ self , other : Series [int ] | Series [float ] | Series [ Timedelta ]
341
+ ) -> Series [ Timedelta ] : ...
348
342
def __divmod__ (self , other : timedelta ) -> tuple [int , Timedelta ]: ...
349
343
# Mypy complains Forward operator "<inequality op>" is not callable, so ignore misc
350
344
# for le, lt ge and gt
@@ -358,7 +352,7 @@ class Timedelta(timedelta):
358
352
self , other : np_ndarray [ShapeT , np .timedelta64 ]
359
353
) -> np_ndarray [ShapeT , np .bool_ ]: ...
360
354
@overload
361
- def __le__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ...
355
+ def __le__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ...
362
356
# Override due to more types supported than dt.timedelta
363
357
@overload # type: ignore[override]
364
358
def __lt__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[misc]
@@ -369,7 +363,7 @@ class Timedelta(timedelta):
369
363
self , other : np_ndarray [ShapeT , np .timedelta64 ]
370
364
) -> np_ndarray [ShapeT , np .bool_ ]: ...
371
365
@overload
372
- def __lt__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ...
366
+ def __lt__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ...
373
367
# Override due to more types supported than dt.timedelta
374
368
@overload # type: ignore[override]
375
369
def __ge__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[misc]
@@ -380,7 +374,7 @@ class Timedelta(timedelta):
380
374
self , other : np_ndarray [ShapeT , np .timedelta64 ]
381
375
) -> np_ndarray [ShapeT , np .bool_ ]: ...
382
376
@overload
383
- def __ge__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ...
377
+ def __ge__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ...
384
378
# Override due to more types supported than dt.timedelta
385
379
@overload # type: ignore[override]
386
380
def __gt__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[misc]
@@ -391,7 +385,7 @@ class Timedelta(timedelta):
391
385
self , other : np_ndarray [ShapeT , np .timedelta64 ]
392
386
) -> np_ndarray [ShapeT , np .bool_ ]: ...
393
387
@overload
394
- def __gt__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ...
388
+ def __gt__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ...
395
389
def __hash__ (self ) -> int : ...
396
390
def isoformat (self ) -> str : ...
397
391
def to_numpy (self ) -> np .timedelta64 : ...
0 commit comments