Skip to content

Commit f16deed

Browse files
committed
fix: fix unittest test_missing_tensorrt_package to run only when min_torch is 2.2.0 or above.
1 parent 8489571 commit f16deed

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/tests_pytorch/models/test_torch_tensorrt.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ def test_torch_minimum_version():
2424

2525

2626
@pytest.mark.skipif(_TORCH_TRT_AVAILABLE, reason="Run this test only if tensorrt is not available.")
27+
@RunIf(min_torch="2.2.0")
2728
def test_missing_tensorrt_package():
2829
model = BoringModel()
2930
with pytest.raises(

0 commit comments

Comments
 (0)