FEAT: Remove API layer default for groupby.get_group() (modin-project#70

) [should upstream]
mvashishtha · Feb 22, 2023 · 8f9608d · 8f9608d
1 parent cf4b4b0
commit 8f9608d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/modin/core/execution/client/query_compiler.py b/modin/core/execution/client/query_compiler.py
@@ -949,6 +949,7 @@ def forwarding_method(self, by, *args, **kwargs):
         "tail",
         "nth",
         "ngroup",
+        "get_group",
         "std",
         "sem",
         "rank",

diff --git a/modin/pandas/groupby.py b/modin/pandas/groupby.py
@@ -672,7 +672,13 @@ def var(self, ddof=1):
         )
 
     def get_group(self, name, obj=None):
-        return self._default_to_pandas(lambda df: df.get_group(name, obj=obj))
+        return self._check_index(
+            self._wrap_aggregation(
+                qc_method=type(self._query_compiler).groupby_get_group,
+                numeric_only=False,
+                agg_kwargs=dict(name=name, obj=obj),
+            )
+        )
 
     def __len__(self):
         return len(self.indices)
@@ -1333,6 +1339,7 @@ def nsmallest(self, n=5, keep="first"):
             )
         )
 
+
 if IsExperimental.get():
     from modin.experimental.cloud.meta_magic import make_wrapped_class