Skip to content

Commit 49e590a

Browse files
Use static shape information about inputs
1 parent f22edc0 commit 49e590a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sunode/wrappers/as_pytensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def read_dict(vals, name=None):
4646
tensor, dim_names = vals
4747
else:
4848
try:
49-
tensor, dim_names = vals, pt.as_tensor_variable(vals, dtype="float64").shape.eval()
49+
tensor, dim_names = vals, pt.as_tensor_variable(vals, dtype="float64").type.shape
5050
except MissingInputError as e:
5151
raise ValueError(
5252
'Shapes of tensors need to be statically '

0 commit comments

Comments
 (0)