[DLMED] add @SkipIfBeforePyTorchVersion((1, 7))

Signed-off-by: Nic Ma <nma@nvidia.com>
Project-MONAI · Feb 2, 2021 · 46b3328 · 46b3328
1 parent e87901a
commit 46b3328
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_handler_classification_saver_dist.py b/tests/test_handler_classification_saver_dist.py
@@ -20,9 +20,10 @@
 from ignite.engine import Engine
 
 from monai.handlers import ClassificationSaver
-from tests.utils import DistCall, DistTestCase
+from tests.utils import DistCall, DistTestCase, SkipIfBeforePyTorchVersion
 
 
+@SkipIfBeforePyTorchVersion((1, 7))
 class DistributedHandlerClassificationSaver(DistTestCase):
     @DistCall(nnodes=1, nproc_per_node=2)
     def test_saved_content(self):