type func arg of core.apply.frame_apply

Author: arw2019

pandas/core/apply.py

@@ -6,7 +6,7 @@
 
 from pandas._config import option_context
 
-from pandas._typing import Axis, FrameOrSeriesUnion
+from pandas._typing import ApplyFuncType, Axis, FrameOrSeriesUnion
 from pandas.util._decorators import cache_readonly
 
 from pandas.core.dtypes.common import (
@@ -27,7 +27,7 @@
 
 def frame_apply(
  obj: "DataFrame",
- func,
+ func: ApplyFuncType,
  axis: Axis = 0,
  raw: bool = False,
  result_type: Optional[str] = None,