Update onnx test skip condition

Summary:
Pull Request resolved: facebookresearch#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

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):