wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-13 17:45:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

[docker] Support multiple node types (#10504)

Browse Source
This commit is contained in:
Ian Rodney
2020-09-02 18:27:59 -07:00
committed by GitHub
parent dc7fe1a4c5
commit b9633a2b67
5 changed files with 152 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/autoscaler/command_runner.py
+2 -3
View File
@@ -705,9 +705,8 @@ class DockerCommandRunner(CommandRunnerInterface):
def run_init(self, *, as_head, file_mounts):
image = self.docker_config.get("image")
if image is None:
image = self.docker_config.get(
f"{'head' if as_head else 'worker'}_image")
image = self.docker_config.get(
f"{'head' if as_head else 'worker'}_image", image)
self._check_docker_installed()
if self.docker_config.get("pull_before_run", True):