diff --git a/python/ray/data/_internal/execution/bulk_executor.py b/python/ray/data/_internal/execution/bulk_executor.py index 8126d7c627eb..fbf515418543 100644 --- a/python/ray/data/_internal/execution/bulk_executor.py +++ b/python/ray/data/_internal/execution/bulk_executor.py @@ -17,6 +17,11 @@ def _naive_task_execute( inputs: List[RefBundle], op: OneToOneOperator ) -> List[RefBundle]: + """Naively execute a 1:1 operation using Ray tasks. + + TODO: This should be reconciled with ComputeStrategy. + """ + @ray.remote(num_returns=2) def transform_one(block: Block) -> (Block, BlockMetadata): [out] = list(op.execute_one([block], {}))