mirror of
https://github.com/wassname/ray.git
synced 2026-10-29 13:30:42 +08:00
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:
1 parent
a82124d304
commit
c8c4832794
1 file changed
+4
-4
@@ -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
|
||||
|
||||
Reference in new issue
Block a user