From 6e7033849eadad34aa7a34547e14ed3f2b9c411c Mon Sep 17 00:00:00 2001 From: Cheng Su Date: Mon, 29 Jan 2024 12:46:22 -0800 Subject: [PATCH] Address comments Signed-off-by: Cheng Su --- .../_internal/execution/operators/task_pool_map_operator.py | 2 +- python/ray/data/_internal/util.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/ray/data/_internal/execution/operators/task_pool_map_operator.py b/python/ray/data/_internal/execution/operators/task_pool_map_operator.py index 01afc6830357..b46a039f44ff 100644 --- a/python/ray/data/_internal/execution/operators/task_pool_map_operator.py +++ b/python/ray/data/_internal/execution/operators/task_pool_map_operator.py @@ -42,7 +42,6 @@ def __init__( or None to use as many tasks as possible. ray_remote_args: Customize the ray remote args for this op's tasks. """ - self._concurrency = concurrency super().__init__( map_transformer, input_op, @@ -51,6 +50,7 @@ def __init__( min_rows_per_bundle, ray_remote_args, ) + self._concurrency = concurrency def _add_bundled_input(self, bundle: RefBundle): # Submit the task as a normal Ray task. diff --git a/python/ray/data/_internal/util.py b/python/ray/data/_internal/util.py index 2d251163f5ce..d5dfedff84fe 100644 --- a/python/ray/data/_internal/util.py +++ b/python/ray/data/_internal/util.py @@ -601,7 +601,7 @@ def get_compute_strategy( raise ValueError( "``concurrency`` is set as a tuple of integers, but ``fn`` " f"is not a callable class: {fn}. Use ``concurrency=n`` to " - "control maximal number of workers to use." + "control maximum number of workers to use." ) else: raise ValueError(