[Docker] Fix Docker Stop on Teardown (ray-project#11400)

bobwenx · Oct 16, 2020 · 6d6a536 · 6d6a536
1 parent 6233cef
commit 6d6a536
Showing 1 changed file with 11 additions and 20 deletions.
diff --git a/python/ray/autoscaler/_private/commands.py b/python/ray/autoscaler/_private/commands.py
@@ -359,26 +359,17 @@ def remaining_nodes():
 
         def run_docker_stop(node, container_name):
             try:
-                updater = NodeUpdaterThread(
-                    node_id=node,
-                    provider_config=config["provider"],
-                    provider=provider,
-                    auth_config=config["auth"],
-                    cluster_name=config["cluster_name"],
-                    file_mounts=config["file_mounts"],
-                    initialization_commands=[],
-                    setup_commands=[],
-                    ray_start_commands=[],
-                    runtime_hash="",
-                    file_mounts_contents_hash="",
-                    is_head_node=False,
-                    docker_config=config.get("docker"))
-                _exec(
-                    updater,
-                    f"docker stop {container_name}",
-                    False,
-                    False,
-                    run_env="host")
+                exec_cluster(
+                    config_file,
+                    cmd=f"docker stop {container_name}",
+                    run_env="host",
+                    screen=False,
+                    tmux=False,
+                    stop=False,
+                    start=False,
+                    override_cluster_name=override_cluster_name,
+                    port_forward=None,
+                    with_output=False)
             except Exception:
                 cli_logger.warning(f"Docker stop failed on {node}")
                 cli_logger.old_warning(logger, f"Docker stop failed on {node}")