ray.rllib.evaluation.worker_set.WorkerSet.foreach_worker_with_id#

WorkerSet.foreach_worker_with_id(func: Callable[[int, EnvRunner], T], *, local_worker: bool = True, healthy_only: bool = False, remote_worker_ids: List[int] = None, timeout_seconds: int | None = None) List[T][source]#

Similar to foreach_worker(), but calls the function with id of the worker too.

Parameters:
  • func – The function to call for each worker (as only arg).

  • local_worker – Whether apply func on local worker too. Default is True.

  • healthy_only – Apply func on known-to-be healthy workers only.

  • remote_worker_ids – Apply func on a selected set of remote workers.

  • timeout_seconds – Time to wait for results. Default is None.

Returns:

The list of return values of all calls to func([worker, id]).