Skip to content

Commit 584e288

Browse files
GH1327 PR feedback
1 parent f26b710 commit 584e288

File tree

5 files changed

+37
-27
lines changed

5 files changed

+37
-27
lines changed

pandas-stubs/core/groupby/groupby.pyi

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -74,9 +74,9 @@ from pandas._typing import (
7474
from pandas.plotting import PlotAccessor
7575

7676
_ResamplerGroupBy: TypeAlias = (
77-
DatetimeIndexResamplerGroupby[NDFrameT]
78-
| PeriodIndexResamplerGroupby[NDFrameT]
79-
| TimedeltaIndexResamplerGroupby[NDFrameT]
77+
DatetimeIndexResamplerGroupby[NDFrameT] # ty: ignore
78+
| PeriodIndexResamplerGroupby[NDFrameT] # ty: ignore
79+
| TimedeltaIndexResamplerGroupby[NDFrameT] # ty: ignore
8080
)
8181

8282
class GroupBy(BaseGroupBy[NDFrameT]):

pandas-stubs/core/indexes/datetimes.pyi

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -104,8 +104,9 @@ class DatetimeIndex(
104104

105105
@overload
106106
def date_range(
107-
end: str | DateAndDatetimeLike,
108-
periods: int,
107+
*,
108+
end: str | DateAndDatetimeLike = ...,
109+
periods: int = ...,
109110
freq: str | timedelta | Timedelta | BaseOffset | None = None,
110111
tz: TimeZones = None,
111112
normalize: bool = False,
@@ -116,7 +117,8 @@ def date_range(
116117
@overload
117118
def date_range(
118119
start: str | DateAndDatetimeLike,
119-
periods: int,
120+
*,
121+
periods: int = ...,
120122
freq: str | timedelta | Timedelta | BaseOffset | None = None,
121123
tz: TimeZones = None,
122124
normalize: bool = False,
@@ -126,8 +128,9 @@ def date_range(
126128
) -> DatetimeIndex: ...
127129
@overload
128130
def date_range(
129-
start: str | DateAndDatetimeLike | None,
130-
end: str | DateAndDatetimeLike | None,
131+
start: str | DateAndDatetimeLike = ...,
132+
end: str | DateAndDatetimeLike = ...,
133+
*,
131134
freq: str | timedelta | Timedelta | BaseOffset | None = None,
132135
tz: TimeZones = None,
133136
normalize: bool = False,
@@ -137,10 +140,10 @@ def date_range(
137140
) -> DatetimeIndex: ...
138141
@overload
139142
def date_range(
140-
start: str | DateAndDatetimeLike,
141-
end: str | DateAndDatetimeLike,
142-
periods: int,
143-
freq: None = None,
143+
start: str | DateAndDatetimeLike = ...,
144+
end: str | DateAndDatetimeLike = ...,
145+
periods: int = ...,
146+
*,
144147
tz: TimeZones = None,
145148
normalize: bool = False,
146149
name: Hashable | None = None,

pandas-stubs/core/indexes/timedeltas.pyi

Lines changed: 13 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -81,19 +81,18 @@ class TimedeltaIndex(
8181

8282
@overload
8383
def timedelta_range(
84-
start: TimedeltaConvertibleTypes,
85-
end: TimedeltaConvertibleTypes,
86-
periods: int,
87-
freq: None = None,
84+
start: TimedeltaConvertibleTypes = ...,
85+
end: TimedeltaConvertibleTypes = ...,
86+
periods: int = ...,
87+
*,
8888
name: Hashable | None = None,
8989
closed: Literal["left", "right"] | None = None,
90-
*,
9190
unit: None | str = ...,
9291
) -> TimedeltaIndex: ...
9392
@overload
9493
def timedelta_range(
95-
start: TimedeltaConvertibleTypes,
96-
periods: int,
94+
start: TimedeltaConvertibleTypes = ...,
95+
periods: int = ...,
9796
freq: None = None,
9897
name: Hashable | None = None,
9998
closed: Literal["left", "right"] | None = None,
@@ -102,31 +101,31 @@ def timedelta_range(
102101
) -> TimedeltaIndex: ...
103102
@overload
104103
def timedelta_range(
105-
start: TimedeltaConvertibleTypes,
106-
periods: int,
104+
start: TimedeltaConvertibleTypes = ...,
105+
*,
106+
periods: int = ...,
107107
freq: str | DateOffset | Timedelta | dt.timedelta | None = None,
108108
name: Hashable | None = None,
109109
closed: Literal["left", "right"] | None = None,
110-
*,
111110
unit: None | str = ...,
112111
) -> TimedeltaIndex: ...
113112
@overload
114113
def timedelta_range(
115-
start: TimedeltaConvertibleTypes,
116-
end: TimedeltaConvertibleTypes,
114+
start: TimedeltaConvertibleTypes = ...,
115+
end: TimedeltaConvertibleTypes = ...,
116+
*,
117117
freq: str | DateOffset | Timedelta | dt.timedelta | None = None,
118118
name: Hashable | None = None,
119119
closed: Literal["left", "right"] | None = None,
120-
*,
121120
unit: None | str = ...,
122121
) -> TimedeltaIndex: ...
123122
@overload
124123
def timedelta_range(
124+
*,
125125
end: TimedeltaConvertibleTypes,
126126
periods: int,
127127
freq: str | DateOffset | Timedelta | dt.timedelta | None = None,
128128
name: Hashable | None = None,
129129
closed: Literal["left", "right"] | None = None,
130-
*,
131130
unit: None | str = ...,
132131
) -> TimedeltaIndex: ...

tests/test_styler.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,6 +143,7 @@ def test_highlight_quantile() -> None:
143143

144144
def test_loader() -> None:
145145
if PD_LTE_23:
146+
# see GH62123 for pandas main repo, type changes in pandas 3.0
146147
check(assert_type(DF.style.loader, PackageLoader), PackageLoader)
147148

148149

tests/test_timefuncs.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1567,7 +1567,7 @@ def test_timedelta_range() -> None:
15671567
day1 = pd.Timedelta(1, unit="D")
15681568
day10 = pd.Timedelta(10, unit="D")
15691569
pd.timedelta_range(
1570-
day1, day10, periods=10, freq="D" # type: ignore[call-overload] # pyright: ignore[reportArgumentType]
1570+
day1, day10, 10, "D" # type: ignore[call-overload] # pyright: ignore[reportArgumentType]
15711571
)
15721572

15731573

@@ -1750,6 +1750,13 @@ def test_creating_date_range() -> None:
17501750
dr = pd.date_range(start="2021-12-01", periods=24, freq="h")
17511751
check(assert_type(dr.strftime("%H:%M:%S"), pd.Index), pd.Index, str)
17521752

1753+
if TYPE_CHECKING_INVALID_USAGE:
1754+
day1 = pd.Timestamp("2023-04-03")
1755+
day10 = pd.Timedelta("2023-04-08")
1756+
pd.date_range(
1757+
day1, day10, 10, "D" # type: ignore[call-overload] # pyright: ignore[reportCallIssue]
1758+
)
1759+
17531760

17541761
def test_timestamp_to_list_add() -> None:
17551762
# https://github.com/microsoft/python-type-stubs/issues/110

0 commit comments

Comments
 (0)