Skip to content

Commit 36dfe7f

Browse files
GH 956 Allowing sequence type in to_timedelta
1 parent e78aaca commit 36dfe7f

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

pandas-stubs/core/tools/timedeltas.pyi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def to_timedelta(
3131
@overload
3232
def to_timedelta(
3333
arg: (
34-
Sequence[float | timedelta]
34+
Sequence[str | float | timedelta]
3535
| list[str | float | timedelta]
3636
| tuple[str | float | timedelta, ...]
3737
| range

tests/test_timefuncs.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -140,6 +140,15 @@ def test_timedelta_series_arithmetic() -> None:
140140
r4: pd.TimedeltaIndex = tds1 / 10.2
141141

142142

143+
def test_timedelta_series_string() -> None:
144+
tds1: pd.TimedeltaIndex = pd.to_timedelta(["1 day"])
145+
td1: pd.Timedelta = pd.Timedelta("2 days")
146+
r1: pd.TimedeltaIndex = tds1 + td1
147+
r2: pd.TimedeltaIndex = tds1 - td1
148+
r3: pd.TimedeltaIndex = tds1 * 4.3
149+
r4: pd.TimedeltaIndex = tds1 / 10.2
150+
151+
143152
def test_timestamp_timedelta_series_arithmetic() -> None:
144153
ts1 = pd.to_datetime(pd.Series(["2022-03-05", "2022-03-06"]))
145154
assert isinstance(ts1.iloc[0], pd.Timestamp)

0 commit comments

Comments
 (0)