From c8c4832794c42ae67f0d57590fdb90a458653d51 Mon Sep 17 00:00:00 2001 From: raoul-khour-ts <69156393+raoul-khour-ts@users.noreply.github.com> Date: Mon, 24 Aug 2020 15:29:55 -0400 Subject: [PATCH] Prevent Local Worker creation from blocking remote worker creation by creating remote workers before local worker (#10245) * create remote workers before local worker * reformatted --- rllib/evaluation/worker_set.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/rllib/evaluation/worker_set.py b/rllib/evaluation/worker_set.py index d79e0c4f1..ffa10a2e8 100644 --- a/rllib/evaluation/worker_set.py +++ b/rllib/evaluation/worker_set.py @@ -64,15 +64,15 @@ class WorkerSet: trainer_config, {"tf_session_args": trainer_config["local_tf_session_args"]}) + # Create a number of remote workers + self._remote_workers = [] + self.add_workers(num_workers) + # Always create a local worker self._local_worker = self._make_worker(RolloutWorker, env_creator, self._policy_class, 0, self._local_config) - # Create a number of remote workers - self._remote_workers = [] - self.add_workers(num_workers) - def local_worker(self) -> RolloutWorker: """Return the local rollout worker.""" return self._local_worker