cover edge case (single row)

rapidsai · Feb 26, 2021 · e68d4a2 · e68d4a2
1 parent 56dcbe4
commit e68d4a2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/dask_cudf/dask_cudf/core.py b/python/dask_cudf/dask_cudf/core.py
@@ -488,6 +488,8 @@ def _finalize_var(vals):
                 (local_name, p, depth - 1) for p in range(lstart, lstop)
             ]
             dsk[(local_name, group, depth)] = (_aggregate_var, node_list)
+    if height == 1:
+        group = depth = 0
     dsk[(name, 0)] = (_finalize_var, (local_name, group, depth))
 
     graph = HighLevelGraph.from_collections(name, dsk, dependencies=[num, ddf])