Skip to content

Commit a488907

Browse files
committed
fix(comment): #1274 (review)
1 parent e2b1901 commit a488907

File tree

3 files changed

+11
-4
lines changed

3 files changed

+11
-4
lines changed

pandas-stubs/core/series.pyi

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1892,6 +1892,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
18921892
self: Series[Timestamp],
18931893
other: (
18941894
timedelta
1895+
| Sequence[timedelta]
18951896
| np.timedelta64
18961897
| np_ndarray_td
18971898
| TimedeltaIndex
@@ -2169,6 +2170,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
21692170
self: Series[Timestamp],
21702171
other: (
21712172
timedelta
2173+
| Sequence[timedelta]
21722174
| np.timedelta64
21732175
| np_ndarray_td
21742176
| TimedeltaIndex

tests/series/arithmetic/timestamp/test_add.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -92,16 +92,19 @@ def test_add_py_sequence() -> None:
9292

9393
if TYPE_CHECKING_INVALID_USAGE:
9494
_0 = left + s # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
95+
# Series[Timestamp] + Sequence[timedelta] should work, see pandas-dev/pandas#62353
9596
_a = left + d # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
9697

9798
_1 = s + left # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
9899
_b = d + left # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
99100

100-
left.add(s) # type: ignore[call-overload] # pyright: ignore[reportArgumentType,reportCallIssue]
101-
left.add(d) # type: ignore[call-overload] # pyright: ignore[reportArgumentType,reportCallIssue]
101+
if TYPE_CHECKING_INVALID_USAGE:
102+
left.add(s) # type: ignore[arg-type] # pyright: ignore[reportArgumentType,reportCallIssue]
103+
left.add(d)
102104

103-
left.radd(s) # type: ignore[call-overload] # pyright: ignore[reportArgumentType,reportCallIssue]
104-
left.radd(d) # type: ignore[call-overload] # pyright: ignore[reportArgumentType,reportCallIssue]
105+
if TYPE_CHECKING_INVALID_USAGE:
106+
left.radd(s) # type: ignore[arg-type] # pyright: ignore[reportArgumentType,reportCallIssue]
107+
left.radd(d)
105108

106109

107110
def test_add_numpy_array() -> None:

tests/series/arithmetic/timestamp/test_sub.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,9 @@ def test_sub_py_sequence() -> None:
8686
d = [timedelta(seconds=1)]
8787

8888
if TYPE_CHECKING_INVALID_USAGE:
89+
# Series[Timestamp] - Sequence[timestamp] should work, see pandas-dev/pandas#62353
8990
_0 = left - s # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
91+
# Series[Timestamp] - Sequence[timedelta] should work, see pandas-dev/pandas#62353
9092
_a = left - d # type: ignore[operator] # pyright: ignore[reportOperatorIssue]
9193

9294
_1 = s - left # type: ignore[operator] # pyright: ignore[reportOperatorIssue]

0 commit comments

Comments
 (0)