Prevent Local Worker creation from blocking remote worker creation by creating remote workers before local worker (#10245)

* create remote workers before local worker

* reformatted
This commit is contained in:
raoul-khour-ts
2020-08-24 12:29:55 -07:00
committed by GitHub
parent a82124d304
commit c8c4832794
+4 -4
View File
@@ -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