groupby

Author: cmp0xff

pandas-stubs/core/groupby/generic.pyi

@@ -70,11 +70,11 @@ class SeriesGroupBy(GroupBy[Series[S2]], Generic[S2, ByT]):
     @overload
     def aggregate(
         self,
-        func: Callable[[Series], C2] | Callable[Concatenate[Series, ...], C2],
+        func: Callable[[Series], C2],
         /,
         *args: Any,
-        engine: WindowingEngine = ...,
-        engine_kwargs: WindowingEngineKwargs = ...,
+        engine: WindowingEngine = None,
+        engine_kwargs: WindowingEngineKwargs = None,
         **kwargs: Any,
     ) -> Series[C2]: ...
     @overload
@@ -83,8 +83,8 @@ class SeriesGroupBy(GroupBy[Series[S2]], Generic[S2, ByT]):
         func: list[AggFuncTypeBase[...]],
         /,
         *args: Any,
-        engine: WindowingEngine = ...,
-        engine_kwargs: WindowingEngineKwargs = ...,
+        engine: WindowingEngine = None,
+        engine_kwargs: WindowingEngineKwargs = None,
         **kwargs: Any,
     ) -> DataFrame: ...
     @overload
@@ -93,8 +93,8 @@ class SeriesGroupBy(GroupBy[Series[S2]], Generic[S2, ByT]):
         func: AggFuncTypeBase[...] | None = ...,
         /,
         *args: Any,
-        engine: WindowingEngine = ...,
-        engine_kwargs: WindowingEngineKwargs = ...,
+        engine: WindowingEngine = None,
+        engine_kwargs: WindowingEngineKwargs = None,
         **kwargs: Any,
     ) -> Series: ...
     agg = aggregate
@@ -104,8 +104,8 @@ class SeriesGroupBy(GroupBy[Series[S2]], Generic[S2, ByT]):
         func: Callable[Concatenate[Series[S2], P], Series[S3]],
         /,
         *args: Any,
-        engine: WindowingEngine = ...,
-        engine_kwargs: WindowingEngineKwargs = ...,
+        engine: WindowingEngine = None,
+        engine_kwargs: WindowingEngineKwargs = None,
         **kwargs: Any,
     ) -> Series[S3]: ...
     @overload
@@ -249,17 +249,17 @@ class DataFrameGroupBy(GroupBy[DataFrame], Generic[ByT, _TT]):
         self,
         func: Literal["size"],
         *args: Any,
-        engine: WindowingEngine = ...,
-        engine_kwargs: WindowingEngineKwargs = ...,
+        engine: WindowingEngine = None,
+        engine_kwargs: WindowingEngineKwargs = None,
         **kwargs: Any,
     ) -> Series: ...
     @overload
     def aggregate(
         self,
         func: AggFuncTypeFrame[..., Any] | None = ...,
         *args: Any,
-        engine: WindowingEngine = ...,
-        engine_kwargs: WindowingEngineKwargs = ...,
+        engine: WindowingEngine = None,
+        engine_kwargs: WindowingEngineKwargs = None,
         **kwargs: Any,
     ) -> DataFrame: ...
     @overload
@@ -275,8 +275,8 @@ class DataFrameGroupBy(GroupBy[DataFrame], Generic[ByT, _TT]):
         self,
         func: Callable[Concatenate[DataFrame, P], DataFrame],
         *args: Any,
-        engine: WindowingEngine = ...,
-        engine_kwargs: WindowingEngineKwargs = ...,
+        engine: WindowingEngine = None,
+        engine_kwargs: WindowingEngineKwargs = None,
         **kwargs: Any,
     ) -> DataFrame: ...
     @overload

tests/frame/test_groupby.py

@@ -300,8 +300,6 @@ def test_types_groupby_agg() -> None:
     def wrapped_min(x: pd.Series[S1]) -> S1:
         return x.min()
 
-    # TODO: facebook/pyrefly#0
-    # pyrefly: ignore[assert-type]
     check(assert_type(df.groupby("col1")["col3"].agg(min), pd.Series), pd.Series)
     check(
         assert_type(df.groupby("col1")["col3"].agg([min, max]), pd.DataFrame),
@@ -712,8 +710,6 @@ def test_getattr_and_dataframe_groupby() -> None:
     df = pd.DataFrame(
         data={"col1": [1, 1, 2], "col2": [3, 4, 5], "col3": [0, 1, 0], 0: [-1, -1, -1]}
     )
-    # TODO: facebook/pyrefly#0
-    # pyrefly: ignore[assert-type]
     check(assert_type(df.groupby("col1").col3.agg(min), pd.Series), pd.Series)
     check(
         assert_type(df.groupby("col1").col3.agg([min, max]), pd.DataFrame), pd.DataFrame