needs keys to pass to __getitem__

rapidsai · Dec 20, 2024 · e663c01 · e663c01
1 parent bfc4e59
commit e663c01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cudf/cudf/core/groupby/groupby.py b/python/cudf/cudf/core/groupby/groupby.py
@@ -3071,7 +3071,7 @@ def _reduce_numeric_only(self, op: str):
 
     def __getitem__(self, key):
         return self.obj[key].groupby(
-            by=self.grouping,
+            by=self.grouping.keys,
             dropna=self._dropna,
             sort=self._sort,
             group_keys=self._group_keys,