skip cpu-only test

Signed-off-by: Wenqi Li <[email protected]>
Project-MONAI · Oct 13, 2023 · 7306433 · 7306433
1 parent 958f002
commit 7306433
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_bundle_download.py b/tests/test_bundle_download.py
@@ -28,6 +28,7 @@
     assert_allclose,
     command_line_tests,
     skip_if_downloading_fails,
+    skip_if_no_cuda,
     skip_if_quick,
 )
 
@@ -156,6 +157,7 @@ def test_monaihosting_source_download_bundle(self, bundle_files, bundle_name, ve
                     self.assertTrue(os.path.exists(file_path))
 
 
+@skip_if_no_cuda
 class TestLoad(unittest.TestCase):
     @parameterized.expand([TEST_CASE_4])
     @skip_if_quick