ray.rllib.evaluation.worker_set.WorkerSet.foreach_worker_with_id
ray.rllib.evaluation.worker_set.WorkerSet.foreach_worker_with_id#
- WorkerSet.foreach_worker_with_id(func: Callable[[int, ray.rllib.evaluation.rollout_worker.RolloutWorker], ray.rllib.evaluation.worker_set.T], *, local_worker=True, healthy_only=False, remote_worker_ids: Optional[List[int]] = None, timeout_seconds: Optional[int] = None) List[ray.rllib.evaluation.worker_set.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 active workers only. By default this will apply func on all workers regardless of their states.
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])
.