Skip to content

Commit

Permalink
Update onnx test skip condition
Browse files Browse the repository at this point in the history
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
  • Loading branch information
Eric Zhang authored and facebook-github-bot committed May 29, 2024
1 parent 39d77a7 commit 0df2d73
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_export_onnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 0df2d73

Please sign in to comment.