lint

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

Author: alexeykudinkin

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

@@ -564,9 +564,9 @@ def __init__(self, min_rows_per_bundle: Optional[int]):
                 bundle up to this target, but only exceed it if not doing so would
                 result in an empty bundle.
         """
-        assert min_rows_per_bundle is None or min_rows_per_bundle >= 0, (
-            "Min rows per bundle has to be non-negative"
-        )
+        assert (
+            min_rows_per_bundle is None or min_rows_per_bundle >= 0
+        ), "Min rows per bundle has to be non-negative"
 
         self._min_rows_per_bundle = min_rows_per_bundle
         self._bundle_buffer: List[RefBundle] = []

python/ray/data/_internal/logical/rules/operator_fusion.py

@@ -179,7 +179,9 @@ def _can_fuse(self, down_op: PhysicalOperator, up_op: PhysicalOperator) -> bool:
             (
                 isinstance(up_logical_op, AbstractMap)
                 and isinstance(down_logical_op, AbstractMap)
-                and self._can_fuse_map_ops(up_logical_op, down_logical_op, up_op.data_context)
+                and self._can_fuse_map_ops(
+                    up_logical_op, down_logical_op, up_op.data_context
+                )
             )
             or (
                 isinstance(up_logical_op, AbstractMap)
@@ -309,8 +311,7 @@ def _get_fused_map_operator(
 
         # Derive min num rows per input bundle
         min_rows_per_bundled_input = self._derive_bundle_min_num_rows(
-            down_logical_op,
-            up_logical_op
+            down_logical_op, up_logical_op
         )
 
         target_max_block_size = self._get_merged_target_max_block_size(
@@ -488,10 +489,7 @@ def _can_fuse_map_ops(
 
         # Do not fuse read op with downstream map op in case when downstream has
         # `min_rows_per_input_bundle` specified (to avoid reducing reading parallelism)
-        if (
-            upstream_op.is_read_op()
-            and ds_bundle_min_rows_req is not None
-        ):
+        if upstream_op.is_read_op() and ds_bundle_min_rows_req is not None:
             return False
 
         # Avoid fusing operators in cases when doing so would lead to a dramatic parallelism

python/ray/data/_internal/plan.py

@@ -259,7 +259,7 @@ def generate_logical_plan_string(
                         chars_left_for_col_name = max(
                             SCHEMA_LINE_CHAR_LIMIT - len(shortened_suffix),
                             MIN_FIELD_LENGTH,
-                            )
+                        )
                         col_str = (
                             f"{col_str[:chars_left_for_col_name]}{shortened_suffix}"
                         )