Skip to content

Commit f6dd4a2

Browse files
Include unit argument in timedelta_range type stub (#1271)
Update stub for timedelta_range to include unit
1 parent 5b38eb5 commit f6dd4a2

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

pandas-stubs/core/indexes/timedeltas.pyi

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,4 +83,6 @@ def timedelta_range(
8383
freq: str | DateOffset | Timedelta | dt.timedelta | None = ...,
8484
name: Hashable | None = ...,
8585
closed: Literal["left", "right"] | None = ...,
86+
*,
87+
unit: None | str = ...,
8688
) -> TimedeltaIndex: ...

tests/test_timefuncs.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1309,6 +1309,18 @@ def test_timedelta_range() -> None:
13091309
),
13101310
pd.TimedeltaIndex,
13111311
)
1312+
check(
1313+
assert_type(
1314+
pd.timedelta_range(
1315+
pd.Timedelta(1, unit="D"),
1316+
pd.Timedelta(10, unit="D"),
1317+
periods=10,
1318+
unit="s",
1319+
),
1320+
pd.TimedeltaIndex,
1321+
),
1322+
pd.TimedeltaIndex,
1323+
)
13121324

13131325

13141326
def test_dateoffset_freqstr() -> None:

0 commit comments

Comments
 (0)