fixes #2821

Signed-off-by: Wenqi Li <wenqil@nvidia.com>
Project-MONAI · Aug 26, 2021 · f425787 · f425787
1 parent 935b1d8
commit f425787
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_gaussian_filter.py b/tests/test_gaussian_filter.py
@@ -16,7 +16,7 @@
 from parameterized import parameterized
 
 from monai.networks.layers import GaussianFilter
-from tests.utils import skip_if_quick
+from tests.utils import SkipIfBeforePyTorchVersion, skip_if_quick
 
 TEST_CASES = [[{"type": "erf", "gt": 2.0}], [{"type": "scalespace", "gt": 3.0}], [{"type": "sampled", "gt": 5.0}]]
 TEST_CASES_GPU = [
@@ -85,6 +85,7 @@ def code_to_run(self, input_args):
             )
 
     @parameterized.expand(TEST_CASES + TEST_CASES_GPU + TEST_CASES_3d)
+    @SkipIfBeforePyTorchVersion((1, 7))
     def test_train_quick(self, input_args):
         self.code_to_run(input_args)