[Data] Avoid warning of compute argument in add/select/drop_columns() (…

…ray-project#43392)

This PR is to fix the bug when user uses `add/select/drop_columns` without `compute`/`concurrency` API. Before this PR, it prints out a deprecation warning by mistake. This PR is to avoid printing out the warning.

Signed-off-by: Cheng Su <[email protected]>

python/ray/data/dataset.py

@@ -646,11 +646,6 @@ def add_column(
  ray_remote_args: Additional resource requirements to request from
  ray (e.g., num_gpus=1 to request GPUs for the map tasks).
  """
- compute = get_compute_strategy(
- fn,
- compute=compute,
- concurrency=concurrency,
- )
 
  def process_batch(batch: "pandas.DataFrame") -> "pandas.DataFrame":
  batch.loc[:, col] = fn(batch)
@@ -663,6 +658,7 @@ def process_batch(batch: "pandas.DataFrame") -> "pandas.DataFrame":
  process_batch,
  batch_format="pandas", # TODO(ekl) we should make this configurable.
  compute=compute,
+ concurrency=concurrency,
  zero_copy_batch=False,
  **ray_remote_args,
  )
@@ -713,16 +709,12 @@ def drop_columns(
  def fn(batch):
  return batch.drop(columns=cols)
 
- compute = get_compute_strategy(
- fn,
- compute=compute,
- concurrency=concurrency,
- )
  return self.map_batches(
  fn,
  batch_format="pandas",
  zero_copy_batch=True,
  compute=compute,
+ concurrency=concurrency,
  **ray_remote_args,
  )
 
@@ -772,16 +764,12 @@ def select_columns(
  def fn(batch):
  return BlockAccessor.for_block(batch).select(columns=cols)
 
- compute = get_compute_strategy(
- fn,
- compute=compute,
- concurrency=concurrency,
- )
  return self.map_batches(
  fn,
  batch_format="pandas",
  zero_copy_batch=True,
  compute=compute,
+ concurrency=concurrency,
  **ray_remote_args,
  )