Be less strict about static shape information for Scan's inner-nonseqs

brandonwillard · brandonwillard · commit 6307c50eb1a5 · 2022-08-24T01:08:41.000-05:00
diff --git a/aesara/scan/op.py b/aesara/scan/op.py
@@ -1210,7 +1210,7 @@ def make_node(self, *inputs):
         ):
             outer_nonseq = copy_var_format(_outer_nonseq, as_var=inner_nonseq)
             new_inputs.append(outer_nonseq)
-            if not outer_nonseq.type.in_same_class(inner_nonseq.type):
+            if not inner_nonseq.type.is_super(outer_nonseq.type):
                 raise ValueError(
                     (
                         f"Argument {outer_nonseq} given to the scan node is not"

Original file line number	Diff line number	Diff line change
`@@ -1210,7 +1210,7 @@ def make_node(self, *inputs):`
`1210`	`1210`	`):`
`1211`	`1211`	`outer_nonseq = copy_var_format(_outer_nonseq, as_var=inner_nonseq)`
`1212`	`1212`	`new_inputs.append(outer_nonseq)`
`1213`		`- if not outer_nonseq.type.in_same_class(inner_nonseq.type):`
	`1213`	`+ if not inner_nonseq.type.is_super(outer_nonseq.type):`
`1214`	`1214`	`raise ValueError(`
`1215`	`1215`	`(`
`1216`	`1216`	`f"Argument {outer_nonseq} given to the scan node is not"`