Update pydra/engine/state.py

Co-authored-by: Chris Markiewicz <[email protected]>

Author: tclose

pydra/engine/state.py

@@ -1155,7 +1155,7 @@ def _processing_terms(self, term, previous_states_ind):
             # checking if the term is in inner_inputs
             if term 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)
+                inner_len = [shape[-1]] * prod(shape[:-1])
                 # this come from the previous node
                 outer_ind = self.inner_inputs[term].ind_l
                 var_ind_out = itertools.chain.from_iterable(