Use static shape information about inputs

pymc-devs · Aug 18, 2023 · 49e590a · 49e590a
1 parent f22edc0
commit 49e590a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sunode/wrappers/as_pytensor.py b/sunode/wrappers/as_pytensor.py
@@ -46,7 +46,7 @@ def read_dict(vals, name=None):
                 tensor, dim_names = vals
             else:
                 try:
-                    tensor, dim_names = vals, pt.as_tensor_variable(vals, dtype="float64").shape.eval()
+                    tensor, dim_names = vals, pt.as_tensor_variable(vals, dtype="float64").type.shape
                 except MissingInputError as e:
                     raise ValueError(
                         'Shapes of tensors need to be statically '