fix rechunk

dask · Nov 14, 2024 · 4d6826a · 4d6826a
1 parent f30fd2d
commit 4d6826a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/distributed/shuffle/_rechunk.py b/distributed/shuffle/_rechunk.py
@@ -121,7 +121,7 @@
 
 import dask
 import dask.config
-from dask._task_spec import Task, TaskRef
+from dask._task_spec import Task, TaskRef, parse_input
 from dask.highlevelgraph import HighLevelGraph
 from dask.layers import Layer
 from dask.tokenize import tokenize
@@ -756,7 +756,9 @@ def partial_concatenate(
             rec_cat_arg[old_partial_index] = TaskRef((input_name,) + old_global_index)
 
     concat_task = Task(
-        (rechunk_name(token),) + global_new_index, concatenate3, rec_cat_arg.tolist()
+        (rechunk_name(token),) + global_new_index,
+        concatenate3,
+        parse_input(rec_cat_arg.tolist()),
     )
     dsk[concat_task.key] = concat_task
     return dsk