File tree Expand file tree Collapse file tree 2 files changed +9
-2
lines changed Expand file tree Collapse file tree 2 files changed +9
-2
lines changed Original file line number Diff line number Diff line change @@ -188,7 +188,6 @@ from pandas.core.dtypes.dtypes import CategoricalDtype
188
188
from pandas .plotting import PlotAccessor
189
189
190
190
_T_COMPLEX = TypeVar ("_T_COMPLEX" , bound = complex )
191
- _T_TIMESTAMP = TypeVar ("_T_TIMESTAMP" , bound = Timestamp )
192
191
193
192
_scalar_timestamp : TypeAlias = date | datetime | np .datetime64
194
193
_vector_timestamp : TypeAlias = (
@@ -2052,7 +2051,14 @@ class Series(IndexOpsMixin[S1], NDFrame):
2052
2051
) -> Series [Timestamp ]: ...
2053
2052
@overload
2054
2053
def __sub__ (
2055
- self : Series [Timestamp ], other : _nonseries_timestamp | Series [_T_TIMESTAMP ]
2054
+ self : Series [Timestamp ], other : _nonseries_timestamp
2055
+ ) -> TimedeltaSeries : ...
2056
+ # The following overload confuses mypy and disable it from recognising
2057
+ # Series[Any] - Series[Any] should give Series[Any]. It will be fixed
2058
+ # after removing TimedeltaSeries.
2059
+ @overload
2060
+ def __sub__ (
2061
+ self : Series [Timestamp ], other : Series [Timestamp ]
2056
2062
) -> TimedeltaSeries : ...
2057
2063
@overload
2058
2064
def __truediv__ (
Original file line number Diff line number Diff line change @@ -1606,6 +1606,7 @@ def test_series_min_max_sub_axis() -> None:
1606
1606
sm = s1 * s2
1607
1607
sd = s1 / s2
1608
1608
check (assert_type (sa , pd .Series ), pd .Series )
1609
+ # Will be fixed after removing TimedeltaSeries, see Series.__sub__ in series.pyi
1609
1610
check (assert_type (ss , pd .Series ), pd .Series ) # type: ignore[assert-type]
1610
1611
check (assert_type (sm , pd .Series ), pd .Series )
1611
1612
check (assert_type (sd , pd .Series ), pd .Series )
You can’t perform that action at this time.
0 commit comments