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)