fix Project-MONAI#8012

Signed-off-by: YunLiu <[email protected]>
KumoLiu · Aug 13, 2024 · 0e347f9 · 0e347f9
1 parent 7a6f680
commit 0e347f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_vit.py b/tests/test_vit.py
@@ -106,7 +106,7 @@ def test_ill_arg(
             )
 
     @parameterized.expand(TEST_CASE_Vit[:1])
-    @SkipIfBeforePyTorchVersion((1, 9))
+    @SkipIfBeforePyTorchVersion((2, 0))
     def test_script(self, input_param, input_shape, _):
         net = ViT(**(input_param))
         net.eval()