Fixes query 15 column renaming

rapidsai · Mar 15, 2021 · 3faf6d3 · 3faf6d3
1 parent f97a7cf
commit 3faf6d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gpu_bdb/queries/q15/gpu_bdb_query_15.py b/gpu_bdb/queries/q15/gpu_bdb_query_15.py
@@ -124,8 +124,8 @@ def main(client, config):
     regression_groups = agg_df.groupby(["cat"]).agg(
         {"x": ["count", "sum"], "xx": ["sum"], "xy": ["sum"], "y": ["count", "sum"]}
     )
-    regression_groups = regression_groups.rename(
-        columns={
+    regression_groups.columns = regression_groups.columns.map(
+        {
             ("x", "count"): "count_x",
             ("x", "sum"): "sum_x",
             ("xx", "sum"): "sum_xx",