Set ONNX version in test

pobonomo · pobonomo · commit 069eb8b50509 · 2025-10-30T13:59:10.000+01:00
diff --git a/tests/test_onnx/test_onnx_exceptions.py b/tests/test_onnx/test_onnx_exceptions.py
@@ -18,7 +18,8 @@ def test_unsupported_op(self):
         graph = helper.make_graph(
             nodes=[node], name="BadGraph", inputs=[X], outputs=[Y]
         )
-        model = helper.make_model(graph)
+        model = helper.make_model(graph, opset_imports=[helper.make_opsetid("", 18)])
+        model.ir_version = 9
         onnx.checker.check_model(model)
 
         example = np.zeros((1, 4), dtype=float)
diff --git a/tests/test_onnx/test_onnx_formulations.py b/tests/test_onnx/test_onnx_formulations.py
@@ -85,7 +85,8 @@ def build_simple_mlp_onnx(
             initializer=[init_W1, init_b1, init_W2, init_b2],
         )
 
-    model = helper.make_model(graph)
+    model = helper.make_model(graph, opset_imports=[helper.make_opsetid("", 18)])
+    model.ir_version = 9
     onnx.checker.check_model(model)
     return model
 

Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,8 @@ def test_unsupported_op(self):`
`18`	`18`	`graph = helper.make_graph(`
`19`	`19`	`nodes=[node], name="BadGraph", inputs=[X], outputs=[Y]`
`20`	`20`	`)`
`21`		`- model = helper.make_model(graph)`
	`21`	`+ model = helper.make_model(graph, opset_imports=[helper.make_opsetid("", 18)])`
	`22`	`+ model.ir_version = 9`
`22`	`23`	`onnx.checker.check_model(model)`
`23`	`24`
`24`	`25`	`example = np.zeros((1, 4), dtype=float)`
Original file line number	Diff line number	Diff line change
`@@ -85,7 +85,8 @@ def build_simple_mlp_onnx(`
`85`	`85`	`initializer=[init_W1, init_b1, init_W2, init_b2],`
`86`	`86`	`)`
`87`	`87`
`88`		`- model = helper.make_model(graph)`
	`88`	`+ model = helper.make_model(graph, opset_imports=[helper.make_opsetid("", 18)])`
	`89`	`+ model.ir_version = 9`
`89`	`90`	`onnx.checker.check_model(model)`
`90`	`91`	`return model`
`91`	`92`