* test for raft

modelscope · Dec 30, 2024 · 2025a2e · 2025a2e
1 parent 8981b63
commit 2025a2e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/tests/ops/filter/test_video_motion_score_raft_filter.py b/tests/ops/filter/test_video_motion_score_raft_filter.py
@@ -10,7 +10,7 @@
 
 # skip due to conflicts when run lazy_load in multiprocessing in librosa
 # tests passed locally.
-@SKIPPED_TESTS.register_module()
+# @SKIPPED_TESTS.register_module()
 class VideoMotionScoreRaftFilterTest(DataJuicerTestCaseBase):
 
     data_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), '..',
@@ -26,6 +26,7 @@ def _run_helper(self, op, source_list, target_list, np=1):
                                          column=[{}] * dataset.num_rows)
         dataset = dataset.map(op.compute_stats, num_proc=np)
         dataset = dataset.filter(op.process, num_proc=np)
+        print(dataset[Fields.stats])
         dataset = dataset.select_columns(column_names=[op.video_key])
         res_list = dataset.to_list()
         self.assertEqual(res_list, target_list)

diff --git a/tests/ops/mapper/test_nlpcda_zh_mapper.py b/tests/ops/mapper/test_nlpcda_zh_mapper.py
@@ -9,13 +9,10 @@
 
 # Skip tests for this OP in the GitHub actions due to unknown UnicodeEncodeError
 # These tests have been tested locally.
-# @SKIPPED_TESTS.register_module()
+@SKIPPED_TESTS.register_module()
 class NlpaugEnMapperTest(DataJuicerTestCaseBase):
 
     def setUp(self):
-        import sys
-        # set the default encoding to utf-8 explicitly
-        sys.stdout.reconfigure(encoding='utf-8')
         self.samples = Dataset.from_dict({
             'text': ['这里一共有5种不同的数据增强方法', '这是不带数字的测试样例'],
             'meta': ['meta information', 'meta information without numbers'],