TYP: narrow SeriesApply func type

Author: iasoon

pandas/core/apply.py

@@ -1049,7 +1049,7 @@ class SeriesApply(NDFrameApply):
     def __init__(
         self,
         obj: Series,
-        func: AggFuncType,
+        func: AggFuncTypeBase | list[AggFuncTypeBase],
         convert_dtype: bool,
         args,
         kwargs,

pandas/core/series.py

@@ -32,7 +32,7 @@
 )
 from pandas._libs.lib import no_default
 from pandas._typing import (
-    AggFuncType,
+    AggFuncTypeBase,
     ArrayLike,
     Axis,
     Dtype,
@@ -4439,7 +4439,11 @@ def any(
         axis=_shared_doc_kwargs["axis"],
     )
     def transform(
-        self, func: AggFuncType, axis: Axis = 0, *args, **kwargs
+        self,
+        func: AggFuncTypeBase | list[AggFuncTypeBase],
+        axis: Axis = 0,
+        *args,
+        **kwargs,
     ) -> DataFrame | Series:
         # Validate axis argument
         self._get_axis_number(axis)
@@ -4450,7 +4454,7 @@ def transform(
 
     def apply(
         self,
-        func: AggFuncType,
+        func: AggFuncTypeBase | list[AggFuncTypeBase],
         convert_dtype: bool = True,
         args: tuple[Any, ...] = (),
         **kwargs,