From 3faf6d3bc0e278cb1dd66440241d99e6b6143641 Mon Sep 17 00:00:00 2001 From: Chris Jarrett Date: Mon, 15 Mar 2021 11:16:29 -0700 Subject: [PATCH] Fixes query 15 column renaming --- gpu_bdb/queries/q15/gpu_bdb_query_15.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gpu_bdb/queries/q15/gpu_bdb_query_15.py b/gpu_bdb/queries/q15/gpu_bdb_query_15.py index b0ccb678..7699d087 100755 --- 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",