add pytest.skip

open-mmlab · Aug 6, 2021 · 552773b · 552773b
1 parent 513c351
commit 552773b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_data/test_datasets/test_s3dis_dataset.py b/tests/test_data/test_datasets/test_s3dis_dataset.py
@@ -42,6 +42,8 @@ def test_getitem():
 
 
 def test_evaluate():
+    if not torch.cuda.is_available():
+        pytest.skip()
     from mmdet3d.core.bbox.structures import DepthInstance3DBoxes
     root_path = './tests/data/s3dis'
     ann_file = './tests/data/s3dis/s3dis_infos.pkl'