Skip to content

Commit fa92585

Browse files
xcz011WillAyd
authored andcommitted
Add type hint for (core.arrays).ranges (pandas-dev#26936)
1 parent 58cbf81 commit fa92585

File tree

1 file changed

+15
-4
lines changed

1 file changed

+15
-4
lines changed

pandas/core/arrays/_ranges.py

+15-4
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,19 @@
33
(and possibly TimedeltaArray/PeriodArray)
44
"""
55

6+
from typing import Tuple
7+
68
import numpy as np
79

810
from pandas._libs.tslibs import OutOfBoundsDatetime, Timestamp
911

10-
from pandas.tseries.offsets import Tick, generate_range
12+
from pandas.tseries.offsets import DateOffset, Tick, generate_range
1113

1214

13-
def generate_regular_range(start, end, periods, freq):
15+
def generate_regular_range(start: Timestamp,
16+
end: Timestamp,
17+
periods: int,
18+
freq: DateOffset) -> Tuple[np.ndarray, str]:
1419
"""
1520
Generate a range of dates with the spans between dates described by
1621
the given `freq` DateOffset.
@@ -79,7 +84,10 @@ def generate_regular_range(start, end, periods, freq):
7984
return values, tz
8085

8186

82-
def _generate_range_overflow_safe(endpoint, periods, stride, side='start'):
87+
def _generate_range_overflow_safe(endpoint: int,
88+
periods: int,
89+
stride: int,
90+
side: str = 'start') -> int:
8391
"""
8492
Calculate the second endpoint for passing to np.arange, checking
8593
to avoid an integer overflow. Catch OverflowError and re-raise
@@ -146,7 +154,10 @@ def _generate_range_overflow_safe(endpoint, periods, stride, side='start'):
146154
return _generate_range_overflow_safe(midpoint, remaining, stride, side)
147155

148156

149-
def _generate_range_overflow_safe_signed(endpoint, periods, stride, side):
157+
def _generate_range_overflow_safe_signed(endpoint: int,
158+
periods: int,
159+
stride: int,
160+
side: str) -> int:
150161
"""
151162
A special case for _generate_range_overflow_safe where `periods * stride`
152163
can be calculated without overflowing int64 bounds.

0 commit comments

Comments
 (0)