From b755bb2b6382025e0ac71b34b6480ece0380b320 Mon Sep 17 00:00:00 2001 From: Charles Blackmon-Luca <20627856+charlesbluca@users.noreply.github.com> Date: Tue, 2 Nov 2021 11:06:12 -0700 Subject: [PATCH] Use unique names for null/non-null groupby columns --- dask_sql/physical/utils/groupby.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/dask_sql/physical/utils/groupby.py b/dask_sql/physical/utils/groupby.py index 5a4e78246..33479bd20 100644 --- a/dask_sql/physical/utils/groupby.py +++ b/dask_sql/physical/utils/groupby.py @@ -23,6 +23,9 @@ def get_groupby_with_nulls_cols( is_null_column = ~(group_column.isnull()) non_nan_group_column = group_column.fillna(0) + # split_out doesn't work if both columns have the same name + is_null_column.name = f"{is_null_column.name}_{new_temporary_column(df)}" + group_columns_and_nulls += [is_null_column, non_nan_group_column] if not group_columns_and_nulls: