Update onnx test skip condition

Eric Zhang · facebook-github-bot · commit 0df2d73d0013 · 2024-05-29T11:42:23.000-07:00
Summary: Pull Request resolved: #5297 onnx test was being skipped by checking version == 1.16.0. But onnx released version 1.16.1 which still fails the test but no longer skips. Now, the test is skipped for 1.16.1 as well as any future versions Reviewed By: wat3rBro Differential Revision: D57917730 fbshipit-source-id: c40c9b17aab9560ebfd495d4b6d7d64cc77f5a09
diff --git a/tests/test_export_onnx.py b/tests/test_export_onnx.py
@@ -98,8 +98,8 @@ def inference_func(model, image):
         )
 
     @unittest.skipIf(
-        version.Version(onnx.version.version) == version.Version("1.16.0"),
-        "This test fails on ONNX Runtime 1.16",
+        version.Version(onnx.version.version) >= version.Version("1.16.0"),
+        "This test fails on ONNX Runtime >= 1.16",
     )
     def testKeypointHead(self):
         class M(torch.nn.Module):

Original file line number	Diff line number	Diff line change
`@@ -98,8 +98,8 @@ def inference_func(model, image):`
`98`	`98`	`)`
`99`	`99`
`100`	`100`	`@unittest.skipIf(`
`101`		`- version.Version(onnx.version.version) == version.Version("1.16.0"),`
`102`		`- "This test fails on ONNX Runtime 1.16",`
	`101`	`+ version.Version(onnx.version.version) >= version.Version("1.16.0"),`
	`102`	`+ "This test fails on ONNX Runtime >= 1.16",`
`103`	`103`	`)`
`104`	`104`	`def testKeypointHead(self):`
`105`	`105`	`class M(torch.nn.Module):`