@@ -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