Skip to content

Commit 54c5e9c

Browse files
authored
Merge pull request #14 from hartikainen/add-run
Add cluster name to ray exec helper command
2 parents 2d404e1 + 0556783 commit 54c5e9c

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

python/ray/autoscaler/commands.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,8 @@ def get_or_create_head_node(config, config_file, no_restart, restart_only,
193193
monitor_str = "docker exec {} /bin/sh -c {}".format(
194194
config["docker"]["container_name"], quote(monitor_str))
195195
print("To monitor auto-scaling activity, you can run:\n\n"
196-
" ray exec {} {}\n".format(config_file, quote(monitor_str)))
196+
" ray exec {} {} --cluster-name={}\n".format(
197+
config_file, quote(monitor_str), quote(config["cluster_name"])))
197198
print("To login to the cluster, run:\n\n"
198199
" ssh -i {} {}@{}\n".format(config["auth"]["ssh_private_key"],
199200
config["auth"]["ssh_user"],
@@ -202,7 +203,7 @@ def get_or_create_head_node(config, config_file, no_restart, restart_only,
202203

203204
def attach_cluster(config_file, start, override_cluster_name):
204205
"""Attaches to a screen for the specified cluster.
205-
206+
206207
Arguments:
207208
config_file: path to the cluster yaml
208209
start: whether to start the cluster if it isn't up
@@ -215,7 +216,7 @@ def attach_cluster(config_file, start, override_cluster_name):
215216

216217
def exec_cluster(config_file, cmd, screen, stop, start, override_cluster_name):
217218
"""Runs a command on the specified cluster.
218-
219+
219220
Arguments:
220221
config_file: path to the cluster yaml
221222
cmd: command to run

0 commit comments

Comments
 (0)