Fix unrelated tests/networks/test_convert_to_onnx.py

garciadias · garciadias · commit 93f16dba61f3 · 2025-02-19T10:25:03.000Z
There was a problem with this test after the else statement was removed, leading to a onnx_model not defined error
diff --git a/tests/networks/test_convert_to_onnx.py b/tests/networks/test_convert_to_onnx.py
@@ -64,7 +64,7 @@ def test_unet(self, device, use_trace, use_ort):
                 rtol=rtol,
                 atol=atol,
             )
-        self.assertTrue(isinstance(onnx_model, onnx.ModelProto))
+            self.assertTrue(isinstance(onnx_model, onnx.ModelProto))
 
     @parameterized.expand(TESTS_ORT)
     @SkipIfBeforePyTorchVersion((1, 12))

Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,7 @@ def test_unet(self, device, use_trace, use_ort):`
`64`	`64`	`rtol=rtol,`
`65`	`65`	`atol=atol,`
`66`	`66`	`)`
`67`		`- self.assertTrue(isinstance(onnx_model, onnx.ModelProto))`
	`67`	`+ self.assertTrue(isinstance(onnx_model, onnx.ModelProto))`
`68`	`68`
`69`	`69`	`@parameterized.expand(TESTS_ORT)`
`70`	`70`	`@SkipIfBeforePyTorchVersion((1, 12))`