Skip to content

Commit

Permalink
FEAT: Remove API layer default for groupby.get_group() (modin-project#70
Browse files Browse the repository at this point in the history
) [should upstream] [core]
  • Loading branch information
pyrito authored and vnlitvinov committed Feb 27, 2023
1 parent ce0e9bc commit ba9fac1
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion modin/pandas/groupby.py
Original file line number Diff line number Diff line change
Expand Up @@ -667,7 +667,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)
Expand Down Expand Up @@ -1327,6 +1333,7 @@ def nsmallest(self, n=5, keep="first"):
)
)


if IsExperimental.get():
from modin.experimental.cloud.meta_magic import make_wrapped_class

Expand Down

0 comments on commit ba9fac1

Please sign in to comment.