Skip to content

Commit be6c87d

Browse files
committed
TYP: Imrprove timedelta_range
1 parent e7d5294 commit be6c87d

File tree

1 file changed

+16
-3
lines changed

1 file changed

+16
-3
lines changed

pandas-stubs/core/indexes/timedeltas.pyi

Lines changed: 16 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,10 @@
1-
from typing import overload
1+
from typing import (
2+
Hashable,
3+
Literal,
4+
overload,
5+
)
26

7+
from pandas import DateOffset
38
from pandas.core.indexes.accessors import TimedeltaIndexProperties
49
from pandas.core.indexes.datetimelike import DatetimeTimedeltaMixin
510
from pandas.core.indexes.datetimes import DatetimeIndex
@@ -9,7 +14,10 @@ from pandas._libs import (
914
Timedelta,
1015
Timestamp,
1116
)
12-
from pandas._typing import num
17+
from pandas._typing import (
18+
TimedeltaConvertibleTypes,
19+
num,
20+
)
1321

1422
class TimedeltaIndex(DatetimeTimedeltaMixin, TimedeltaIndexProperties):
1523
def __new__(
@@ -42,5 +50,10 @@ class TimedeltaIndex(DatetimeTimedeltaMixin, TimedeltaIndexProperties):
4250
def to_series(self, index=..., name=...) -> TimedeltaSeries: ...
4351

4452
def timedelta_range(
45-
start=..., end=..., periods=..., freq=..., name=..., closed=...
53+
start: TimedeltaConvertibleTypes = ...,
54+
end: TimedeltaConvertibleTypes = ...,
55+
periods: int | None = ...,
56+
freq: str | DateOffset | None = ...,
57+
name: Hashable | None = ...,
58+
closed: Literal["left", "right"] | None = ...,
4659
) -> TimedeltaIndex: ...

0 commit comments

Comments
 (0)