Use static shape information about inputs

Author: michaelosthege

sunode/wrappers/as_pytensor.py

@@ -45,12 +45,11 @@ def read_dict(vals, name=None):
             if isinstance(vals, tuple):
                 tensor, dim_names = vals
             else:
-                try:
-                    tensor, dim_names = vals, pt.as_tensor_variable(vals, dtype="float64").shape.eval()
-                except MissingInputError as e:
+                tensor, dim_names = vals, pt.as_tensor_variable(vals, dtype="float64").type.shape
+                if any(d is None for d in dim_names):
                     raise ValueError(
-                        'Shapes of tensors need to be statically '
-                        'known or given explicitly.') from e
+                        'Shapes of tensors need to be statically known or given explicitly.'
+                    )
             if isinstance(dim_names, (str, int)):
                 dim_names = (dim_names,)
             tensor = pt.as_tensor_variable(tensor, dtype="float64")