Skip to content

Commit

Permalink
Better fix
Browse files Browse the repository at this point in the history
  • Loading branch information
dcherian committed Nov 5, 2022
1 parent 3b79f6e commit 5cd7bcb
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion flox/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -1350,7 +1350,10 @@ def dask_groupby_agg(
aggregate=partial(aggregate, expected_groups=index, reindex=True),
)
)
groups_.append(cohort)
# This is done because pandas promotes to 64-bit types when an Index is created
# So we use the index to generate the return value for consistency with "map-reduce"
# This is important on windows
groups_.append(index.values)

reduced = dask.array.concatenate(reduced_, axis=-1)
groups = (np.concatenate(groups_),)
Expand Down

0 comments on commit 5cd7bcb

Please sign in to comment.