fix(pandas): compatibility with 2.1 groupby behavior

ibis-project · Aug 30, 2023 · ab3fc9e · ab3fc9e
1 parent ec0514b
commit ab3fc9e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ibis/backends/pandas/execution/generic.py b/ibis/backends/pandas/execution/generic.py
@@ -588,7 +588,10 @@ def execute_aggregation_dataframe(
             )
             for key in op.by
         ]
-        source = data.groupby(grouping_keys, group_keys=False)
+        source = data.groupby(
+            grouping_keys[0] if len(grouping_keys) == 1 else grouping_keys,
+            group_keys=False,
+        )
     else:
         source = data