diff --git a/tests/networks/nets/test_transchex.py b/tests/networks/nets/test_transchex.py index f10c914f0d..1816bc2dd8 100644 --- a/tests/networks/nets/test_transchex.py +++ b/tests/networks/nets/test_transchex.py @@ -18,7 +18,7 @@ from monai.networks import eval_mode from monai.networks.nets.transchex import Transchex -from tests.test_utils import skip_if_quick +from tests.test_utils import skip_if_downloading_fails, skip_if_quick TEST_CASE_TRANSCHEX = [] for drop_out in [0.4]: @@ -49,7 +49,8 @@ class TestTranschex(unittest.TestCase): @parameterized.expand(TEST_CASE_TRANSCHEX) def test_shape(self, input_param, expected_shape): - net = Transchex(**input_param) + with skip_if_downloading_fails(): + net = Transchex(**input_param) with eval_mode(net): result = net(torch.randint(2, (2, 512)), torch.randint(2, (2, 512)), torch.randn((2, 3, 224, 224))) self.assertEqual(result.shape, expected_shape)