Tidying up

Signed-off-by: Alexey Kudinkin <ak@anyscale.com>

Author: alexeykudinkin

python/ray/data/_internal/execution/operators/base_physical_operator.py

@@ -119,7 +119,7 @@ def internal_queue_num_blocks(self) -> int:
         return sum(len(bundle.block_refs) for bundle in self._input_buffer)
 
     def internal_queue_num_bytes(self) -> int:
-        return sum([bundle.size_bytes() for bundle in self._input_buffer])
+        return sum(bundle.size_bytes() for bundle in self._input_buffer)
 
     def all_inputs_done(self) -> None:
         ctx = TaskContext(

python/ray/data/_internal/execution/operators/zip_operator.py

@@ -80,7 +80,7 @@ def num_output_rows_total(self) -> Optional[int]:
         return num_rows
 
     def internal_queue_num_blocks(self) -> int:
-        return sum(len(bundle) for buf in self._input_buffers for bundle in buf)
+        return sum(len(bundle.block_refs) for buf in self._input_buffers for bundle in buf)
 
     def internal_queue_num_bytes(self) -> int:
         return sum(bundle.size_bytes() for buf in self._input_buffers for bundle in buf)