Skip to content

Commit

Permalink
Address comments
Browse files Browse the repository at this point in the history
Signed-off-by: Cheng Su <[email protected]>
  • Loading branch information
c21 committed Jan 29, 2024
1 parent 20f185b commit 6e70338
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand All @@ -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.
Expand Down
2 changes: 1 addition & 1 deletion python/ray/data/_internal/util.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit 6e70338

Please sign in to comment.