fix: #718 mean

Author: cmp0xff

pandas-stubs/core/series.pyi

@@ -205,7 +205,7 @@ _nonseries_timedelta: TypeAlias = _scalar_timedelta | _vector_timedelta
 _nonseries_int: TypeAlias = int | np_ndarray_anyint | Sequence[int]
 
 _T_INT = TypeVar("_T_INT", bound=int)
-_T_STAMP = TypeVar("_T_STAMP", bound=Timestamp)
+_T_STAMP_AND_DELTA = TypeVar("_T_STAMP_AND_DELTA", bound=Timestamp | Timedelta)
 
 class _iLocIndexerSeries(_iLocIndexer, Generic[S1]):
     # get item
@@ -1620,15 +1620,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     ) -> Never: ...
     @overload
     def __add__(
-        self: Series[Timestamp],
-        other: (
-            Timedelta
-            | Series[Timedelta]
-            | TimedeltaIndex
-            | np.timedelta64
-            | timedelta
-            | BaseOffset
-        ),
+        self: Series[Timestamp], other: _nonseries_timedelta | Series[Timedelta]
     ) -> Series[Timestamp]: ...
     @overload
     def __add__(self: Series[Timedelta], other: Period) -> PeriodSeries: ...
@@ -1713,7 +1705,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __or__(self, other: _nonseries_int | Series[int]) -> Series[int]: ...
     @overload
     def __radd__(
-        self: Series[Timestamp], other: Series[Timedelta] | np.timedelta64 | timedelta
+        self: Series[Timestamp], other: _nonseries_timedelta | Series[Timedelta]
     ) -> Series[Timestamp]: ...
     @overload
     def __radd__(
@@ -1734,14 +1726,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __rdivmod__(self, other: num | _ListLike | Series[S1]) -> Series[S1]: ...  # type: ignore[override] # pyright: ignore[reportIncompatibleMethodOverride]
     @overload
     def __rfloordiv__(
-        self: Series[Timedelta],
-        other: (
-            timedelta
-            | Series[Timedelta]
-            | np.timedelta64
-            | TimedeltaIndex
-            | Sequence[timedelta]
-        ),
+        self: Series[Timedelta], other: _nonseries_timedelta | Series[Timedelta]
     ) -> Series[int]: ...
     @overload
     def __rfloordiv__(self, other: num | _ListLike | Series[S1]) -> Series[S1]: ...
@@ -1764,16 +1749,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __rsub__(self, other: num | _ListLike | Series[S1]) -> Series: ...
     @overload
     def __rtruediv__(
-        self: Series[Timedelta],
-        other: (
-            timedelta
-            | Timedelta
-            | Sequence[Timedelta]
-            | Series[Timedelta]
-            | np.timedelta64
-            | TimedeltaIndex
-            | Sequence[timedelta]
-        ),
+        self: Series[Timedelta], other: _nonseries_timedelta | Series[Timedelta]
     ) -> Series[float]: ...
     @overload
     def __rtruediv__(self, other: num | _ListLike | Series[S1] | Path) -> Series: ...
@@ -1791,15 +1767,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     ) -> Series[int]: ...
     @overload
     def __sub__(
-        self: Series[Timestamp],
-        other: (
-            Timedelta
-            | Series[Timedelta]
-            | TimedeltaIndex
-            | np.timedelta64
-            | timedelta
-            | BaseOffset
-        ),
+        self: Series[Timestamp], other: _nonseries_timedelta | Series[Timedelta]
     ) -> Series[Timestamp]: ...
     @overload
     def __sub__(
@@ -1816,21 +1784,12 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __sub__(self, other: num | _ListLike | Series) -> Series: ...
     @overload
     def __truediv__(
-        self: Series[_T_STAMP],
+        self: Series[_T_STAMP_AND_DELTA],
         other: float | Series[int] | Series[float] | Sequence[float],
     ) -> Series[Timestamp]: ...
     @overload
     def __truediv__(
-        self: Series[Timedelta],
-        other: (
-            timedelta
-            | Timedelta
-            | Sequence[Timedelta]
-            | Series[Timedelta]
-            | np.timedelta64
-            | TimedeltaIndex
-            | Sequence[timedelta]
-        ),
+        self: Series[Timedelta], other: _nonseries_timedelta | Series[Timedelta]
     ) -> Series[float]: ...
     @overload
     def __truediv__(
@@ -2023,25 +1982,16 @@ class Series(IndexOpsMixin[S1], NDFrame):
         numeric_only: _bool = ...,
         **kwargs: Any,
     ) -> S1: ...
-    # @overload
-    # def mean(
-    #     self: Series[Timestamp],
-    #     axis: AxisIndex | None = ...,
-    #     skipna: _bool = ...,
-    #     level: None = ...,
-    #     numeric_only: _bool = ...,
-    #     **kwargs: Any,
-    # ) -> Timestamp: ...
-    # @overload
-    # def mean(
-    #     self: Series[Timedelta],
-    #     axis: AxisIndex | None = ...,
-    #     skipna: _bool = ...,
-    #     level: None = ...,
-    #     numeric_only: _bool = ...,
-    #     **kwargs: Any,
-    # ) -> Timedelta: ...
-    # @overload
+    @overload
+    def mean(
+        self: Series[_T_STAMP_AND_DELTA],
+        axis: AxisIndex | None = ...,
+        skipna: _bool = ...,
+        level: None = ...,
+        numeric_only: _bool = ...,
+        **kwargs: Any,
+    ) -> _T_STAMP_AND_DELTA: ...
+    @overload
     def mean(
         self,
         axis: AxisIndex | None = ...,

tests/test_timefuncs.py

@@ -400,10 +400,7 @@ def test_series_dt_accessors() -> None:
         upper="2.3.99",
     ):
         check(
-            assert_type(
-                s0.dt.to_pydatetime(),
-                np.ndarray,
-            ),
+            assert_type(s0.dt.to_pydatetime(), np.ndarray),
             np.ndarray if PD_LTE_23 else pd.Series,
             dt.datetime,
         )
@@ -598,13 +595,7 @@ def test_series_dt_accessors() -> None:
         "The behavior of TimedeltaProperties.to_pytimedelta is deprecated",
         lower="2.3.99",
     ):
-        check(
-            assert_type(
-                s2.dt.to_pytimedelta(),
-                np.ndarray,
-            ),
-            np.ndarray,
-        )
+        check(assert_type(s2.dt.to_pytimedelta(), np.ndarray), np.ndarray)
     check(assert_type(s2.dt.total_seconds(), "pd.Series[float]"), pd.Series, float)
     check(assert_type(s2.dt.unit, TimeUnit), str)
     check(
@@ -1497,7 +1488,7 @@ def test_timedeltaseries_add_timestampseries() -> None:
 
 def test_mean_median_std() -> None:
     s = pd.Series([pd.Timedelta("1 ns"), pd.Timedelta("2 ns"), pd.Timedelta("3 ns")])
-    # check(assert_type(s.mean(), pd.Timedelta), pd.Timedelta)
+    check(assert_type(s.mean(), pd.Timedelta), pd.Timedelta)
     check(assert_type(s.median(), pd.Timedelta), pd.Timedelta)
     check(assert_type(s.std(), pd.Timedelta), pd.Timedelta)
 
@@ -1508,7 +1499,7 @@ def test_mean_median_std() -> None:
             pd.Timestamp("2021-01-03"),
         ]
     )
-    # check(assert_type(s2.mean(), pd.Timestamp), pd.Timestamp)
+    check(assert_type(s2.mean(), pd.Timestamp), pd.Timestamp)
     check(assert_type(s2.median(), pd.Timestamp), pd.Timestamp)
     check(assert_type(s2.std(), pd.Timedelta), pd.Timedelta)