Add pandas.core.groupby.generic classes module setting

Author: ofsouzap

pandas/core/groupby/generic.py

@@ -32,6 +32,7 @@
     Appender,
     Substitution,
     doc,
+    set_module,
 )
 from pandas.util._exceptions import find_stack_level
 
@@ -108,6 +109,7 @@
 ScalarResult = TypeVar("ScalarResult")
 
 
+@set_module("pandas")
 class NamedAgg(NamedTuple):
     """
     Helper for column specific aggregation with control over output column names.
@@ -142,6 +144,7 @@ class NamedAgg(NamedTuple):
     aggfunc: AggScalar
 
 
+@set_module("pandas")
 class SeriesGroupBy(GroupBy[Series]):
     def _wrap_agged_manager(self, mgr: Manager) -> Series:
         out = self.obj._constructor_from_mgr(mgr, axes=mgr.axes)
@@ -1555,6 +1558,7 @@ def unique(self) -> Series:
         return result
 
 
+@set_module("pandas")
 class DataFrameGroupBy(GroupBy[DataFrame]):
     _agg_examples_doc = dedent(
         """

pandas/tests/api/test_api.py

@@ -416,3 +416,6 @@ def test_set_module():
     assert pd.Period.__module__ == "pandas"
     assert pd.Timestamp.__module__ == "pandas"
     assert pd.Timedelta.__module__ == "pandas"
+    assert pd.NamedAgg.__module__ == "pandas"
+    assert pd.SeriesGroupBy.__module__ == "pandas"
+    assert pd.DataFrameGroupBy.__module__ == "pandas"