Remove pdb breakpoint and make single-op-splits robust to 1d shape

nipype · Jun 6, 2024 · 901a99e · 901a99e
1 parent e52e32b
commit 901a99e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pydra/engine/state.py b/pydra/engine/state.py
@@ -1074,10 +1074,9 @@ def _single_op_splits(self, op_single):
         )
         val_ind = range(reduce(lambda x, y: x * y, shape))
         if op_single in self.inner_inputs:
-            if len(shape) == 1:
-                breakpoint()
             # TODO: have to be changed if differ length
-            inner_len = [shape[-1]] * reduce(lambda x, y: x * y, shape[:-1])
+            inner_len = [shape[-1]] * reduce(lambda x, y: x * y, shape[:-1], 1)
+
             # this come from the previous node
             outer_ind = self.inner_inputs[op_single].ind_l
             op_out = itertools.chain.from_iterable(