From ff5590e0aca48d83354f8ea70ec851c0b8c2c5f2 Mon Sep 17 00:00:00 2001 From: YunLiu <55491388+KumoLiu@users.noreply.github.com> Date: Tue, 8 Apr 2025 23:32:22 +0800 Subject: [PATCH] fix #8415 Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com> --- tests/networks/nets/test_transchex.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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)