TYP: add SeriesAggFuncType alias

Author: iasoon

pandas/_typing.py

@@ -159,10 +159,10 @@
 
 # types of `func` kwarg for DataFrame.aggregate and Series.aggregate
 AggFuncTypeBase = Union[Callable, str]
-AggFuncTypeDict = Dict[Hashable, Union[AggFuncTypeBase, List[AggFuncTypeBase]]]
+SeriesAggFuncType = Union[AggFuncTypeBase, List[AggFuncTypeBase]]
+AggFuncTypeDict = Dict[Hashable, SeriesAggFuncType]
 AggFuncType = Union[
-    AggFuncTypeBase,
-    List[AggFuncTypeBase],
+    SeriesAggFuncType,
     AggFuncTypeDict,
 ]
 AggObjType = Union[

pandas/core/apply.py

@@ -32,6 +32,7 @@
     AggObjType,
     Axis,
     NDFrameT,
+    SeriesAggFuncType,
     npt,
 )
 from pandas.util._decorators import cache_readonly
@@ -1049,7 +1050,7 @@ class SeriesApply(NDFrameApply):
     def __init__(
         self,
         obj: Series,
-        func: AggFuncTypeBase | list[AggFuncTypeBase],
+        func: SeriesAggFuncType,
         convert_dtype: bool,
         args,
         kwargs,

pandas/core/series.py

@@ -32,7 +32,6 @@
 )
 from pandas._libs.lib import no_default
 from pandas._typing import (
-    AggFuncTypeBase,
     ArrayLike,
     Axis,
     Dtype,
@@ -43,6 +42,7 @@
     Level,
     NaPosition,
     Renamer,
+    SeriesAggFuncType,
     SingleManager,
     SortKind,
     StorageOptions,
@@ -4440,7 +4440,7 @@ def any(
     )
     def transform(
         self,
-        func: AggFuncTypeBase | list[AggFuncTypeBase],
+        func: SeriesAggFuncType,
         axis: Axis = 0,
         *args,
         **kwargs,
@@ -4454,7 +4454,7 @@ def transform(
 
     def apply(
         self,
-        func: AggFuncTypeBase | list[AggFuncTypeBase],
+        func: SeriesAggFuncType,
         convert_dtype: bool = True,
         args: tuple[Any, ...] = (),
         **kwargs,