bug

xadupre · xadupre · commit 3ce797835a40 · 2025-04-03T13:15:13.000+02:00
diff --git a/_unittests/ut_torch_models/test_test_helpers.py b/_unittests/ut_torch_models/test_test_helpers.py
@@ -64,7 +64,7 @@ def test_validate_model_onnx(self):
             exporter="onnx-dynamo",
             dump_folder="dump_test_validate_model_onnx",
             patch=True,
-            stop_if_static=pv.Version(torch.__version__) > pv.Version("2.6"),
+            stop_if_static=2 if pv.Version(torch.__version__) > pv.Version("2.6") else 0,
         )
         self.assertIsInstance(summary, dict)
         self.assertIsInstance(data, dict)

Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,7 @@ def test_validate_model_onnx(self):`
`64`	`64`	`exporter="onnx-dynamo",`
`65`	`65`	`dump_folder="dump_test_validate_model_onnx",`
`66`	`66`	`patch=True,`
`67`		`- stop_if_static=pv.Version(torch.__version__) > pv.Version("2.6"),`
	`67`	`+ stop_if_static=2 if pv.Version(torch.__version__) > pv.Version("2.6") else 0,`
`68`	`68`	`)`
`69`	`69`	`self.assertIsInstance(summary, dict)`
`70`	`70`	`self.assertIsInstance(data, dict)`