fix

Signed-off-by: Hao Chen <chenh1024@gmail.com>
ray-project · Oct 6, 2023 · 879a088 · 879a088
1 parent 43dfc01
commit 879a088
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/python/ray/data/_internal/execution/operators/map_operator.py b/python/ray/data/_internal/execution/operators/map_operator.py
@@ -260,7 +260,6 @@ def _output_ready_callback(task_index, output: RefBundle):
 
             # Notify output queue that the task has produced an new output.
             self._output_queue.notify_task_output_ready(task_index, output)
-            self._data_tasks[task_index].add_num_output_blocks(len(output.blocks))
 
         def _task_done_callback(task_index):
             self._metrics.on_task_finished(task_index)