agg

Author: cmp0xff

pandas-stubs/core/frame.pyi

@@ -1516,18 +1516,18 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
     @overload
     def aggregate(  # pyright: ignore[reportOverlappingOverload]
         self,
-        func: AggFuncTypeBase | AggFuncTypeDictSeries,
+        func: AggFuncTypeBase | AggFuncTypeDictSeries[Hashable],
         axis: Axis = 0,
         **kwargs: Any,
     ) -> Series: ...
     @overload
     def aggregate(
         self,
-        func: list[AggFuncTypeBase] | AggFuncTypeDictFrame | None = ...,
+        func: list[AggFuncTypeBase] | AggFuncTypeDictFrame[Hashable] | None = ...,
         axis: Axis = 0,
         **kwargs: Any,
     ) -> Self: ...
-    agg = aggregate  # pyright: ignore[reportUnknownVariableType]
+    agg = aggregate
     def transform(
         self,
         func: AggFuncTypeFrame,

pandas-stubs/core/groupby/generic.pyi

@@ -261,7 +261,7 @@ class DataFrameGroupBy(GroupBy[DataFrame], Generic[ByT, _TT]):
     @overload
     def aggregate(
         self,
-        func: AggFuncTypeFrame | None = ...,
+        func: AggFuncTypeFrame[Hashable] | None = ...,
         *args: Any,
         engine: WindowingEngine = ...,
         engine_kwargs: WindowingEngineKwargs = ...,
@@ -270,11 +270,11 @@ class DataFrameGroupBy(GroupBy[DataFrame], Generic[ByT, _TT]):
     @overload
     def aggregate(
         self,
-        func: AggFuncTypeFrame | None = None,
+        func: AggFuncTypeFrame[Hashable] | None = None,
         /,
         **kwargs: Any,
     ) -> DataFrame: ...
-    agg = aggregate  # pyright: ignore[reportUnknownVariableType]
+    agg = aggregate
     @overload
     def transform(
         self,

pandas-stubs/core/series.pyi

@@ -1142,12 +1142,12 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
     @overload
     def aggregate(
         self,
-        func: AggFuncTypeSeriesToFrame = ...,
+        func: AggFuncTypeSeriesToFrame[Hashable] = ...,
         axis: AxisIndex = ...,
         *args: Any,
         **kwargs: Any,
     ) -> Series: ...
-    agg = aggregate  # pyright: ignore[reportUnknownVariableType]
+    agg = aggregate
     @overload
     def transform(
         self,

pandas-stubs/core/window/rolling.pyi

@@ -1,5 +1,6 @@
 from collections.abc import (
     Callable,
+    Hashable,
     Iterator,
 )
 import datetime as dt
@@ -52,18 +53,18 @@ class BaseWindow(SelectionMixin[NDFrameT]):
     @overload
     def aggregate(
         self: BaseWindow[Series],
-        func: AggFuncTypeSeriesToFrame,
+        func: AggFuncTypeSeriesToFrame[Hashable],
         *args: Any,
         **kwargs: Any,
     ) -> DataFrame: ...
     @overload
     def aggregate(
         self: BaseWindow[DataFrame],
-        func: AggFuncTypeFrame,
+        func: AggFuncTypeFrame[Hashable],
         *args: Any,
         **kwargs: Any,
     ) -> DataFrame: ...
-    agg = aggregate  # pyright: ignore[reportUnknownVariableType]
+    agg = aggregate
 
 class BaseWindowGroupby(BaseWindow[NDFrameT]): ...