From ee6be2620d79c1fc3bbee451e204d5342fce4024 Mon Sep 17 00:00:00 2001 From: dcherian Date: Tue, 25 Oct 2022 17:04:30 -0600 Subject: [PATCH] bugfix --- flox/core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flox/core.py b/flox/core.py index 015598185..a51b6ed3f 100644 --- a/flox/core.py +++ b/flox/core.py @@ -1368,7 +1368,7 @@ def dask_groupby_agg( raise ValueError(f"Unknown method={method}.") # extract results from the dict - adjust_chunks = {inds[ax]: lambda: 0 for ax in axis} + adjust_chunks = {inds[ax]: lambda c: 0 for ax in axis} if method == "blockwise" and len(axis) > 1: nblocks = tuple(len(array.chunks[ax]) for ax in axis) group_chunks = np.unravel_index(group_chunks, nblocks)