fix return none in RayExecutor.run_op (#339)

modelscope · Jul 4, 2024 · bfebbde · bfebbde
1 parent 1244d4f
commit bfebbde
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/data_juicer/core/ray_executor.py b/data_juicer/core/ray_executor.py
@@ -155,6 +155,8 @@ def run_op(self, op, op_cfg, dataset):
                     'Ray executor only support Filter and Mapper OPs for '
                     'now')
                 raise NotImplementedError
+
+            return dataset
         except:  # noqa: E722
             logger.error(f'An error occurred during Op [{op_name}].')
             import traceback