Update pydra/engine/state.py

tclose · effigies · web-flow · commit 5c3584f61c78 · 2025-05-06T09:02:54.000+10:00
Co-authored-by: Chris Markiewicz &lt;markiewicz@stanford.edu&gt;
diff --git a/pydra/engine/state.py b/pydra/engine/state.py
@@ -1172,7 +1172,7 @@ def _single_op_splits(self, op_single):
             self.inputs[op_single],
             container_ndim=self.container_ndim_all.get(op_single, 1),
         )
-        val_ind = range(reduce(lambda x, y: x * y, shape, 1))
+        val_ind = range(prod(shape))
         if op_single in self.inner_inputs:
             # TODO: have to be changed if differ length
             inner_len = [shape[-1]] * reduce(lambda x, y: x * y, shape[:-1], 1)

Original file line number	Diff line number	Diff line change
`@@ -1172,7 +1172,7 @@ def _single_op_splits(self, op_single):`
`1172`	`1172`	`self.inputs[op_single],`
`1173`	`1173`	`container_ndim=self.container_ndim_all.get(op_single, 1),`
`1174`	`1174`	`)`
`1175`		`- val_ind = range(reduce(lambda x, y: x * y, shape, 1))`
	`1175`	`+ val_ind = range(prod(shape))`
`1176`	`1176`	`if op_single in self.inner_inputs:`
`1177`	`1177`	`# TODO: have to be changed if differ length`
`1178`	`1178`	`inner_len = [shape[-1]] * reduce(lambda x, y: x * y, shape[:-1], 1)`