wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 12:10:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

[autoscaler] Move attach helper text under exec_cluster (#3920)

Browse Source 
## What do these changes do?
Moves the attach command helper from cli commands to the actual `exec_cluster` function.
This commit is contained in:
Kristian Hartikainen
2019-01-31 17:01:24 -08:00
committed by Richard Liaw
parent 62a0a7bdc7
commit b9eed2e86c
2 changed files with 16 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/scripts/scripts.py
-32
View File
@@ -605,21 +605,6 @@ def submit(cluster_config_file, screen, tmux, stop, start, cluster_name,
exec_cluster(cluster_config_file, cmd, screen, tmux, stop, False,
cluster_name, port_forward)
if tmux or screen:
attach_command_parts = ["ray attach", cluster_config_file]
if cluster_name is not None:
attach_command_parts.append(
"--cluster-name={}".format(cluster_name))
if tmux:
attach_command_parts.append("--tmux")
elif screen:
attach_command_parts.append("--screen")
attach_command = " ".join(attach_command_parts)
attach_info = "Use `{}` to check on command status.".format(
attach_command)
logger.info(attach_info)
@cli.command()
@click.argument("cluster_config_file", required=True, type=str)
@@ -651,26 +636,9 @@ def submit(cluster_config_file, screen, tmux, stop, start, cluster_name,
"--port-forward", required=False, type=int, help="Port to forward.")
def exec_cmd(cluster_config_file, cmd, screen, tmux, stop, start, cluster_name,
port_forward):
assert not (screen and tmux), "Can specify only one of `screen` or `tmux`."
exec_cluster(cluster_config_file, cmd, screen, tmux, stop, start,
cluster_name, port_forward)
if tmux or screen:
attach_command_parts = ["ray attach", cluster_config_file]
if cluster_name is not None:
attach_command_parts.append(
"--cluster-name={}".format(cluster_name))
if tmux:
attach_command_parts.append("--tmux")
elif screen:
attach_command_parts.append("--screen")
attach_command = " ".join(attach_command_parts)
attach_info = "Use `{}` to check on command status.".format(
attach_command)
logger.info(attach_info)
@cli.command()
@click.argument("cluster_config_file", required=True, type=str)