@@ -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 ,
@@ -159,10 +158,7 @@ class Timedelta(timedelta):
159
158
@overload
160
159
def __add__ (self , other : pd .TimedeltaIndex ) -> pd .TimedeltaIndex : ...
161
160
@overload
162
- def __add__ (
163
- self ,
164
- other : TimedeltaSeries ,
165
- ) -> TimedeltaSeries : ...
161
+ def __add__ (self , other : Series [Timedelta ]) -> Series [Timedelta ]: ...
166
162
@overload
167
163
def __add__ (self , other : Series [Timestamp ]) -> Series [Timestamp ]: ...
168
164
@overload
@@ -195,9 +191,7 @@ class Timedelta(timedelta):
195
191
@overload
196
192
def __sub__ (self , other : pd .TimedeltaIndex ) -> TimedeltaIndex : ...
197
193
@overload
198
- def __sub__ (
199
- self , other : TimedeltaSeries | Series [pd .Timedelta ]
200
- ) -> TimedeltaSeries : ...
194
+ def __sub__ (self , other : Series [pd .Timedelta ]) -> Series [pd .Timedelta ]: ...
201
195
@overload
202
196
def __rsub__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> Timedelta : ...
203
197
@overload
@@ -231,9 +225,9 @@ class Timedelta(timedelta):
231
225
self , other : npt .NDArray [np .integer ] | npt .NDArray [np .floating ]
232
226
) -> npt .NDArray [np .timedelta64 ]: ...
233
227
@overload
234
- def __mul__ (self , other : Series [int ]) -> TimedeltaSeries : ...
228
+ def __mul__ (self , other : Series [int ]) -> Series [ Timedelta ] : ...
235
229
@overload
236
- def __mul__ (self , other : Series [float ]) -> TimedeltaSeries : ...
230
+ def __mul__ (self , other : Series [float ]) -> Series [ Timedelta ] : ...
237
231
@overload
238
232
def __mul__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
239
233
@overload
@@ -243,9 +237,9 @@ class Timedelta(timedelta):
243
237
self , other : npt .NDArray [np .floating ] | npt .NDArray [np .integer ]
244
238
) -> npt .NDArray [np .timedelta64 ]: ...
245
239
@overload
246
- def __rmul__ (self , other : Series [int ]) -> TimedeltaSeries : ...
240
+ def __rmul__ (self , other : Series [int ]) -> Series [ Timedelta ] : ...
247
241
@overload
248
- def __rmul__ (self , other : Series [float ]) -> TimedeltaSeries : ...
242
+ def __rmul__ (self , other : Series [float ]) -> Series [ Timedelta ] : ...
249
243
# maybe related to https://github.com/python/mypy/issues/10755
250
244
@overload
251
245
def __rmul__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
@@ -266,11 +260,11 @@ class Timedelta(timedelta):
266
260
@overload
267
261
def __floordiv__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
268
262
@overload
269
- def __floordiv__ (self , other : Series [int ]) -> TimedeltaSeries : ...
263
+ def __floordiv__ (self , other : Series [int ]) -> Series [ Timedelta ] : ...
270
264
@overload
271
- def __floordiv__ (self , other : Series [float ]) -> TimedeltaSeries : ...
265
+ def __floordiv__ (self , other : Series [float ]) -> Series [ Timedelta ] : ...
272
266
@overload
273
- def __floordiv__ (self , other : TimedeltaSeries ) -> Series [int ]: ...
267
+ def __floordiv__ (self , other : Series [ Timedelta ] ) -> Series [int ]: ...
274
268
@overload
275
269
def __floordiv__ (self , other : NaTType | None ) -> float : ...
276
270
@overload
@@ -291,19 +285,19 @@ class Timedelta(timedelta):
291
285
self , other : npt .NDArray [np .integer ] | npt .NDArray [np .floating ]
292
286
) -> npt .NDArray [np .timedelta64 ]: ...
293
287
@overload
294
- def __truediv__ (self , other : TimedeltaSeries ) -> Series [float ]: ...
288
+ def __truediv__ (self , other : Series [ Timedelta ] ) -> Series [float ]: ...
295
289
@overload
296
- def __truediv__ (self , other : Series [int ]) -> TimedeltaSeries : ...
290
+ def __truediv__ (self , other : Series [int ]) -> Series [ Timedelta ] : ...
297
291
@overload
298
- def __truediv__ (self , other : Series [float ]) -> TimedeltaSeries : ...
292
+ def __truediv__ (self , other : Series [float ]) -> Series [ Timedelta ] : ...
299
293
@overload
300
294
def __truediv__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
301
295
def __rtruediv__ (self , other : timedelta | Timedelta | NaTType ) -> float : ...
302
296
# Override due to more types supported than dt.timedelta
303
297
@overload
304
298
def __eq__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
305
299
@overload
306
- def __eq__ (self , other : TimedeltaSeries | Series [pd . Timedelta ]) -> Series [bool ]: ... # type: ignore[overload-overlap]
300
+ def __eq__ (self , other : Series [Timedelta ]) -> Series [bool ]: ... # type: ignore[overload-overlap]
307
301
@overload
308
302
def __eq__ ( # type: ignore[overload-overlap]
309
303
self , other : TimedeltaIndex | npt .NDArray [np .timedelta64 ]
@@ -314,7 +308,7 @@ class Timedelta(timedelta):
314
308
@overload
315
309
def __ne__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
316
310
@overload
317
- def __ne__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ... # type: ignore[overload-overlap]
311
+ def __ne__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ... # type: ignore[overload-overlap]
318
312
@overload
319
313
def __ne__ ( # type: ignore[overload-overlap]
320
314
self , other : TimedeltaIndex | npt .NDArray [np .timedelta64 ]
@@ -327,7 +321,7 @@ class Timedelta(timedelta):
327
321
@overload
328
322
def __mod__ (self , other : float ) -> Timedelta : ...
329
323
@overload
330
- def __mod__ (self , other : Series [int ] | Series [float ]) -> TimedeltaSeries : ...
324
+ def __mod__ (self , other : Series [int ] | Series [float ]) -> Series [ Timedelta ] : ...
331
325
@overload
332
326
def __mod__ (self , other : Index [int ] | Index [float ]) -> TimedeltaIndex : ...
333
327
@overload
@@ -336,8 +330,8 @@ class Timedelta(timedelta):
336
330
) -> npt .NDArray [np .timedelta64 ]: ...
337
331
@overload
338
332
def __mod__ (
339
- self , other : Series [int ] | Series [float ] | TimedeltaSeries
340
- ) -> TimedeltaSeries : ...
333
+ self , other : Series [int ] | Series [float ] | Series [ Timedelta ]
334
+ ) -> Series [ Timedelta ] : ...
341
335
def __divmod__ (self , other : timedelta ) -> tuple [int , Timedelta ]: ...
342
336
# Mypy complains Forward operator "<inequality op>" is not callable, so ignore misc
343
337
# for le, lt ge and gt
@@ -349,7 +343,7 @@ class Timedelta(timedelta):
349
343
self , other : TimedeltaIndex | npt .NDArray [np .timedelta64 ]
350
344
) -> npt .NDArray [np .bool_ ]: ...
351
345
@overload
352
- def __le__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ...
346
+ def __le__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ...
353
347
# Override due to more types supported than dt.timedelta
354
348
@overload # type: ignore[override]
355
349
def __lt__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[misc]
@@ -358,7 +352,7 @@ class Timedelta(timedelta):
358
352
self , other : TimedeltaIndex | npt .NDArray [np .timedelta64 ]
359
353
) -> npt .NDArray [np .bool_ ]: ...
360
354
@overload
361
- def __lt__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ...
355
+ def __lt__ (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 __ge__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[misc]
@@ -367,7 +361,7 @@ class Timedelta(timedelta):
367
361
self , other : TimedeltaIndex | npt .NDArray [np .timedelta64 ]
368
362
) -> npt .NDArray [np .bool_ ]: ...
369
363
@overload
370
- def __ge__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ...
364
+ def __ge__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ...
371
365
# Override due to more types supported than dt.timedelta
372
366
@overload # type: ignore[override]
373
367
def __gt__ (self , other : timedelta | Timedelta | np .timedelta64 ) -> bool : ... # type: ignore[misc]
@@ -376,7 +370,7 @@ class Timedelta(timedelta):
376
370
self , other : TimedeltaIndex | npt .NDArray [np .timedelta64 ]
377
371
) -> npt .NDArray [np .bool_ ]: ...
378
372
@overload
379
- def __gt__ (self , other : TimedeltaSeries | Series [pd .Timedelta ]) -> Series [bool ]: ...
373
+ def __gt__ (self , other : Series [pd .Timedelta ]) -> Series [bool ]: ...
380
374
def __hash__ (self ) -> int : ...
381
375
def isoformat (self ) -> str : ...
382
376
def to_numpy (self ) -> np .timedelta64 : ...
0 commit comments