fix

xadupre · xadupre · commit 98bc77fac050 · 2025-05-08T15:13:41.000+02:00
diff --git a/_unittests/ut_torch_models/test_test_helpers.py b/_unittests/ut_torch_models/test_test_helpers.py
@@ -8,6 +8,7 @@
     ignore_warnings,
     requires_torch,
     requires_experimental,
+    requires_onnxscript,
 )
 from onnx_diagnostic.torch_models.test_helper import (
     get_inputs_for_task,
@@ -88,6 +89,7 @@ def test_validate_model_onnx_dynamo_ir(self):
         )
 
     @requires_torch("2.7")
+    @requires_onnxscript("0.4")
     @hide_stdout()
     @ignore_warnings(FutureWarning)
     def test_validate_model_onnx_dynamo_os_ort(self):

Original file line number	Diff line number	Diff line change
`@@ -8,6 +8,7 @@`
`8`	`8`	`ignore_warnings,`
`9`	`9`	`requires_torch,`
`10`	`10`	`requires_experimental,`
	`11`	`+ requires_onnxscript,`
`11`	`12`	`)`
`12`	`13`	`from onnx_diagnostic.torch_models.test_helper import (`
`13`	`14`	`get_inputs_for_task,`
`@@ -88,6 +89,7 @@ def test_validate_model_onnx_dynamo_ir(self):`
`88`	`89`	`)`
`89`	`90`
`90`	`91`	`@requires_torch("2.7")`
	`92`	`+ @requires_onnxscript("0.4")`
`91`	`93`	`@hide_stdout()`
`92`	`94`	`@ignore_warnings(FutureWarning)`
`93`	`95`	`def test_validate_model_onnx_dynamo_os_ort(self):`