From 5e1c4515d721c4b64c57b3fa718f08b0cb764b46 Mon Sep 17 00:00:00 2001 From: Balaji Veeramani Date: Tue, 21 May 2024 16:06:22 -0700 Subject: [PATCH] Initial commit Signed-off-by: Balaji Veeramani --- python/ray/data/_internal/plan.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/python/ray/data/_internal/plan.py b/python/ray/data/_internal/plan.py index 1df5d90e1d3af..c3a4b06d30b6b 100644 --- a/python/ray/data/_internal/plan.py +++ b/python/ray/data/_internal/plan.py @@ -471,7 +471,6 @@ def _get_num_rows_from_blocks_metadata(self, blocks: BlockList) -> Optional[int] def execute_to_iterator( self, allow_clear_input_blocks: bool = True, - force_read: bool = False, ) -> Tuple[ Iterator[Tuple[ObjectRef[Block], BlockMetadata]], DatasetStats, @@ -484,7 +483,6 @@ def execute_to_iterator( Args: allow_clear_input_blocks: Whether we should try to clear the input blocks for each operator. - force_read: Whether to force the read operator to fully execute. Returns: Tuple of iterator over output blocks and the executor. @@ -496,7 +494,7 @@ def execute_to_iterator( if self.has_computed_output(): return ( self.execute( - allow_clear_input_blocks, force_read + allow_clear_input_blocks, force_read=False ).iter_blocks_with_metadata(), self._snapshot_stats, None,