From e68d4a2333473b935d1d64e9bc0c27e296d516c5 Mon Sep 17 00:00:00 2001 From: rjzamora Date: Thu, 25 Feb 2021 21:04:49 -0800 Subject: [PATCH] cover edge case (single row) --- python/dask_cudf/dask_cudf/core.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/dask_cudf/dask_cudf/core.py b/python/dask_cudf/dask_cudf/core.py index 3cf85e3283f..aa83bad7630 100644 --- 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])