Convert inputs to ndarrays in Blockwise.perform

aesara-devs · Nov 11, 2022 · c9ad602 · c9ad602
1 parent 96853ee
commit c9ad602
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/aesara/tensor/blockwise.py b/aesara/tensor/blockwise.py
@@ -320,6 +320,7 @@ def py_func(*inner_inputs):
             # TODO:This can be avoided by making a single dummy node
             # But will that cover all cases?
             inner_node = self.op.make_node(*inner_inputs)
+            inner_inputs = [np.asarray(i) for i in inner_inputs]
             if isinstance(self.op, DimShuffle):
                 self.op.perform(inner_node, inner_inputs, res, params=None)
             else: