* review for OOM tests

modelscope · Dec 25, 2024 · 38bfd06 · 38bfd06
1 parent c57ca12
commit 38bfd06
Show file tree

Hide file tree

Showing 8 changed files with 8 additions and 8 deletions.
diff --git a/tests/ops/mapper/test_image_captioning_mapper.py b/tests/ops/mapper/test_image_captioning_mapper.py
@@ -11,7 +11,7 @@
 
 # Skip tests for this OP in the GitHub actions due to OOM on the current runner
 # These tests have been tested locally.
-@SKIPPED_TESTS.register_module()
+# @SKIPPED_TESTS.register_module()
 class ImageCaptioningMapperTest(DataJuicerTestCaseBase):
 
     data_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), '..',

diff --git a/tests/ops/mapper/test_image_diffusion_mapper.py b/tests/ops/mapper/test_image_diffusion_mapper.py
@@ -12,7 +12,7 @@
 
 # Skip tests for this OP in the GitHub actions due to OOM on the current runner
 # These tests have been tested locally.
-@SKIPPED_TESTS.register_module()
+# @SKIPPED_TESTS.register_module()
 class ImageDiffusionMapperTest(DataJuicerTestCaseBase):
 
     data_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), '..',

diff --git a/tests/ops/mapper/test_image_tagging_mapper.py b/tests/ops/mapper/test_image_tagging_mapper.py
@@ -10,7 +10,7 @@
 
 # Skip tests for this OP in the GitHub actions due to OOM on the current runner
 # These tests have been tested locally.
-@SKIPPED_TESTS.register_module()
+# @SKIPPED_TESTS.register_module()
 class ImageTaggingMapperTest(DataJuicerTestCaseBase):
     data_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), '..',
                              'data')

diff --git a/tests/ops/mapper/test_video_captioning_from_audio_mapper.py b/tests/ops/mapper/test_video_captioning_from_audio_mapper.py
@@ -11,7 +11,7 @@
 
 # Skip tests for this OP in the GitHub actions due to OOM on the current runner
 # These tests have been tested locally.
-@SKIPPED_TESTS.register_module()
+# @SKIPPED_TESTS.register_module()
 class VideoCaptioningFromAudioMapperTest(DataJuicerTestCaseBase):
 
     data_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), '..',

diff --git a/tests/ops/mapper/test_video_captioning_from_frames_mapper.py b/tests/ops/mapper/test_video_captioning_from_frames_mapper.py
@@ -11,7 +11,7 @@
 
 # Skip tests for this OP in the GitHub actions due to OOM on the current runner
 # These tests have been tested locally.
-@SKIPPED_TESTS.register_module()
+# @SKIPPED_TESTS.register_module()
 class VideoCaptioningFromFramesMapperTest(DataJuicerTestCaseBase):
 
     data_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), '..',

diff --git a/tests/ops/mapper/test_video_captioning_from_summarizer_mapper.py b/tests/ops/mapper/test_video_captioning_from_summarizer_mapper.py
@@ -11,7 +11,7 @@
 
 # Skip tests for this OP in the GitHub actions due to OOM on the current runner
 # These tests have been tested locally.
-@SKIPPED_TESTS.register_module()
+# @SKIPPED_TESTS.register_module()
 class VideoCaptioningFromSummarizerMapperTest(DataJuicerTestCaseBase):
 
     data_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), '..',

diff --git a/tests/ops/mapper/test_video_captioning_from_video_mapper.py b/tests/ops/mapper/test_video_captioning_from_video_mapper.py
@@ -11,7 +11,7 @@
 
 # Skip tests for this OP in the GitHub actions due to OOM on the current runner
 # These tests have been tested locally.
-@SKIPPED_TESTS.register_module()
+# @SKIPPED_TESTS.register_module()
 class VideoCaptioningFromVideoMapperTest(DataJuicerTestCaseBase):
 
     data_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), '..',

diff --git a/tests/ops/mapper/test_video_tagging_from_frames_mapper.py b/tests/ops/mapper/test_video_tagging_from_frames_mapper.py
@@ -11,7 +11,7 @@
 
 # Skip tests for this OP in the GitHub actions due to OOM on the current runner
 # These tests have been tested locally.
-@SKIPPED_TESTS.register_module()
+# @SKIPPED_TESTS.register_module()
 class VideoTaggingFromFramesMapperTest(DataJuicerTestCaseBase):
     data_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), '..',
                              'data')