TimedeltaIndex

Author: cmp0xff

pandas-stubs/core/indexes/timedeltas.pyi

@@ -5,16 +5,14 @@ from collections.abc import (
 import datetime as dt
 from typing import (
     Literal,
+    TypeAlias,
     final,
     overload,
 )
 
 import numpy as np
-from pandas import (
-    Index,
-    Period,
-)
 from pandas.core.indexes.accessors import TimedeltaIndexProperties
+from pandas.core.indexes.base import Index
 from pandas.core.indexes.datetimelike import DatetimeTimedeltaMixin
 from pandas.core.indexes.datetimes import DatetimeIndex
 from pandas.core.indexes.period import PeriodIndex
@@ -26,6 +24,7 @@ from typing_extensions import (
 
 from pandas._libs import Timedelta
 from pandas._libs.tslibs import BaseOffset
+from pandas._libs.tslibs.period import Period
 from pandas._typing import (
     AxesData,
     Frequency,
@@ -40,6 +39,20 @@ from pandas._typing import (
     num,
 )
 
+_NUM_FACTOR: TypeAlias = Just[int] | Just[float] | np.integer | np.floating
+_NUM_FACTORS: TypeAlias = (
+    _NUM_FACTOR
+    | Sequence[_NUM_FACTOR]
+    | np_ndarray_anyint
+    | np_ndarray_float
+    | Index[int]
+    | Index[float]
+)
+_DT_FACTOR: TypeAlias = dt.timedelta | np.timedelta64 | Timedelta
+_DT_FACTORS: TypeAlias = (
+    _DT_FACTOR | Sequence[_DT_FACTOR] | np_ndarray_td | Index[Timedelta]
+)
+
 class TimedeltaIndex(
     DatetimeTimedeltaMixin[Timedelta, np.timedelta64], TimedeltaIndexProperties
 ):
@@ -89,43 +102,20 @@ class TimedeltaIndex(
     @overload  # type: ignore[override]
     def __mul__(self, other: np_ndarray_bool | np_ndarray_complex) -> Never: ...
     @overload
-    def __mul__(
-        self,
-        other: (
-            Just[int]
-            | Just[float]
-            | Sequence[Just[int]]
-            | Sequence[Just[float]]
-            | np_ndarray_anyint
-            | np_ndarray_float
-            | Index[int]
-            | Index[float]
-        ),
-    ) -> Self: ...
+    def __mul__(self, other: _NUM_FACTORS) -> Self: ...
     @overload  # type: ignore[override]
     def __rmul__(self, other: np_ndarray_bool | np_ndarray_complex) -> Never: ...
     @overload
-    def __rmul__(
-        self,
-        other: (
-            Just[int]
-            | Just[float]
-            | Sequence[Just[int]]
-            | Sequence[Just[float]]
-            | np_ndarray_anyint
-            | np_ndarray_float
-            | Index[int]
-            | Index[float]
-        ),
-    ) -> Self: ...
+    def __rmul__(self, other: _NUM_FACTORS) -> Self: ...
     @overload  # type: ignore[override]
-    # pyrefly: ignore  # bad-override
-    def __truediv__(self, other: float | Sequence[float]) -> Self: ...
+    def __truediv__(self, other: _NUM_FACTORS) -> Self: ...
     @overload
     def __truediv__(  # pyright: ignore[reportIncompatibleMethodOverride]
-        self, other: dt.timedelta | Sequence[dt.timedelta]
+        self, other: _DT_FACTORS | Self
+    ) -> Index[float]: ...
+    def __rtruediv__(  # type: ignore[override] # pyright: ignore[reportIncompatibleMethodOverride]
+        self, other: _DT_FACTORS | Self
     ) -> Index[float]: ...
-    def __rtruediv__(self, other: dt.timedelta | Sequence[dt.timedelta]) -> Index[float]: ...  # type: ignore[override]  # pyright: ignore[reportIncompatibleMethodOverride]
     @overload  # type: ignore[override]
     # pyrefly: ignore  # bad-override
     def __floordiv__(self, other: num | Sequence[float]) -> Self: ...