From fd4b2b6bbcd7359435a08be3126dab8dc76dd96c Mon Sep 17 00:00:00 2001 From: Wenqi Li Date: Fri, 15 Oct 2021 09:34:56 +0100 Subject: [PATCH] skip tests for torch 1.5 Signed-off-by: Wenqi Li --- tests/test_integration_fast_train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_integration_fast_train.py b/tests/test_integration_fast_train.py index 09d32f100f..2cf7eee479 100644 --- a/tests/test_integration_fast_train.py +++ b/tests/test_integration_fast_train.py @@ -52,11 +52,12 @@ ToDeviced, ) from monai.utils import set_determinism -from tests.utils import DistTestCase, TimedCall, skip_if_no_cuda, skip_if_quick +from tests.utils import DistTestCase, SkipIfBeforePyTorchVersion, TimedCall, skip_if_no_cuda, skip_if_quick @skip_if_no_cuda @skip_if_quick +@SkipIfBeforePyTorchVersion((1, 7)) class IntegrationFastTrain(DistTestCase): def setUp(self): set_determinism(seed=0)