Skip to content

Commit ec8c20f

Browse files
GH1317 PR feedback
1 parent fbe3373 commit ec8c20f

File tree

2 files changed

+39
-2
lines changed

2 files changed

+39
-2
lines changed

pandas-stubs/core/indexes/interval.pyi

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -322,15 +322,24 @@ def interval_range(
322322
) -> IntervalIndex[Interval[pd.Timestamp]]: ...
323323
@overload
324324
def interval_range(
325-
*,
326325
start: None = None,
326+
*,
327327
end: _TimestampLike,
328328
periods: int | None = None,
329329
freq: Frequency | dt.timedelta | None = None,
330330
name: Hashable = None,
331331
closed: IntervalClosedType = "right",
332332
) -> IntervalIndex[Interval[pd.Timestamp]]: ...
333333
@overload
334+
def interval_range(
335+
start: None,
336+
end: _TimestampLike,
337+
periods: int,
338+
freq: Frequency | dt.timedelta,
339+
name: Hashable = None,
340+
closed: IntervalClosedType = "right",
341+
) -> IntervalIndex[Interval[pd.Timestamp]]: ...
342+
@overload
334343
def interval_range(
335344
start: _TimedeltaLike,
336345
end: _TimedeltaLike | None = None,
@@ -341,8 +350,17 @@ def interval_range(
341350
) -> IntervalIndex[Interval[pd.Timedelta]]: ...
342351
@overload
343352
def interval_range(
344-
*,
353+
start: None,
354+
end: _TimedeltaLike,
355+
periods: int,
356+
freq: Frequency | dt.timedelta,
357+
name: Hashable = None,
358+
closed: IntervalClosedType = "right",
359+
) -> IntervalIndex[Interval[pd.Timedelta]]: ...
360+
@overload
361+
def interval_range(
345362
start: None = None,
363+
*,
346364
end: _TimedeltaLike,
347365
periods: int | None = None,
348366
freq: Frequency | dt.timedelta | None = None,

tests/indexes/test_indexes.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -458,6 +458,25 @@ def test_interval_range() -> None:
458458
pd.IntervalIndex,
459459
pd.Interval,
460460
)
461+
check(
462+
assert_type(
463+
pd.interval_range(None, pd.Timestamp("2020-01-01"), 2, "1D"),
464+
"pd.IntervalIndex[pd.Interval[pd.Timestamp]]",
465+
),
466+
pd.IntervalIndex,
467+
pd.Interval,
468+
)
469+
470+
check(
471+
assert_type(
472+
pd.interval_range(
473+
None, end=pd.Timestamp("2020-01-01"), periods=2, freq="1D"
474+
),
475+
"pd.IntervalIndex[pd.Interval[pd.Timestamp]]",
476+
),
477+
pd.IntervalIndex,
478+
pd.Interval,
479+
)
461480

462481

463482
def test_interval_index_breaks() -> None:

0 commit comments

Comments
 (0)