From bfebbdec4c97d703720f71e68918d4821006e4ad Mon Sep 17 00:00:00 2001
From: Cathy0908 <30484308+Cathy0908@users.noreply.github.com>
Date: Thu, 4 Jul 2024 13:18:22 +0800
Subject: [PATCH] fix return none in RayExecutor.run_op (#339)

---
 data_juicer/core/ray_executor.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/data_juicer/core/ray_executor.py b/data_juicer/core/ray_executor.py
index 1c066ea02..9a0e0835b 100644
--- 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