diff --git a/tests/pipelines/test_referring_video_object_segmentation.py b/tests/pipelines/test_referring_video_object_segmentation.py index 3e81d9c3..4d8206b3 100644 --- a/tests/pipelines/test_referring_video_object_segmentation.py +++ b/tests/pipelines/test_referring_video_object_segmentation.py @@ -14,7 +14,7 @@ class ReferringVideoObjectSegmentationTest(unittest.TestCase, self.task = Tasks.referring_video_object_segmentation self.model_id = 'damo/cv_swin-t_referring_video-object-segmentation' - @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') + @unittest.skip('skip since the model is set to private for now') def test_referring_video_object_segmentation(self): input_location = 'data/test/videos/referring_video_object_segmentation_test_video.mp4' text_queries = [ @@ -31,7 +31,7 @@ class ReferringVideoObjectSegmentationTest(unittest.TestCase, else: raise ValueError('process error') - @unittest.skipUnless(test_level() >= 2, 'skip test in current test level') + @unittest.skip('skip since the model is set to private for now') def test_referring_video_object_segmentation_with_default_task(self): input_location = 'data/test/videos/referring_video_object_segmentation_test_video.mp4' text_queries = [ diff --git a/tests/trainers/test_referring_video_object_segmentation_trainer.py b/tests/trainers/test_referring_video_object_segmentation_trainer.py index c1dc040d..7b03eb4d 100644 --- a/tests/trainers/test_referring_video_object_segmentation_trainer.py +++ b/tests/trainers/test_referring_video_object_segmentation_trainer.py @@ -62,7 +62,7 @@ class TestImageInstanceSegmentationTrainer(unittest.TestCase): self.max_epochs = max_epochs - @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') + @unittest.skip('skip since the model is set to private for now') def test_trainer(self): kwargs = dict( model=self.model_id, @@ -77,7 +77,7 @@ class TestImageInstanceSegmentationTrainer(unittest.TestCase): results_files = os.listdir(trainer.work_dir) self.assertIn(f'{trainer.timestamp}.log.json', results_files) - @unittest.skipUnless(test_level() >= 2, 'skip test in current test level') + @unittest.skip('skip since the model is set to private for now') def test_trainer_with_model_and_args(self): cache_path = snapshot_download(self.model_id)