Minor

hendrikmakait · Aug 13, 2024 · d8a9c8d · d8a9c8d
1 parent dac50a9
commit d8a9c8d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distributed/shuffle/_rechunk.py b/distributed/shuffle/_rechunk.py
@@ -660,7 +660,7 @@ def partial_rechunk(
 
     partial_old = tuple(
         chunk_axis[partial_axis]
-        for partial_axis, chunk_axis in zip(ndpartial.old, chunks)
+        for partial_axis, chunk_axis in zip(ndpartial.old, input_chunks)
     )
     partial_new: ChunkedAxes = tuple(
         chunks[axis_index][ndpartial.new[axis_index]] for axis_index in range(ndim)