cudf compat

narwhals-dev · Feb 21, 2024 · 5b71873 · 5b71873
1 parent 953477c
commit 5b71873
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/narwhals/pandas_like/group_by.py b/narwhals/pandas_like/group_by.py
@@ -105,7 +105,7 @@ def func(df: Any, expr: Expr) -> Any:
             for key in self._keys:
                 if key in res.columns:
                     res = res.drop(columns=[key])
-            res = res.set_axis(expr._output_names, axis=1)
+            res.columns = expr._output_names
             dfs.append(res)
 
         results_keys = dataframe_from_dict(out, implementation=implementation)