Merge b68c5c5 into b13cd06

open-mmlab · Dec 22, 2022 · 72aa051 · 72aa051
2 parents b13cd06 + b68c5c5
commit 72aa051
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_models/test_editors/test_restormer/test_restormer_net.py b/tests/test_models/test_editors/test_restormer/test_restormer_net.py
@@ -1,9 +1,13 @@
 # Copyright (c) OpenMMLab. All rights reserved.
+import pytest
 import torch
 
 from mmedit.models.editors import Restormer
 
 
+@pytest.mark.skipif(
+    torch.__version__ < '1.8.0',
+    reason='skip on torch<1.8 due to unsupported PixelUnShuffle')
 def test_restormer_cpu():
     """Test Restormer."""
 
@@ -79,6 +83,9 @@ def test_restormer_cpu():
     assert output.shape == (1, 3, 16, 16)
 
 
+@pytest.mark.skipif(
+    torch.__version__ < '1.8.0',
+    reason='skip on torch<1.8 due to unsupported PixelUnShuffle')
 def test_restormer_cuda():
     net = Restormer(
         inp_channels=3,