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

[ray] Add --new flag for ray attach (#2973)

Browse Source 
* new flag

* yapf
This commit is contained in:
Eric Liang
2018-09-29 23:04:13 -07:00
committed by GitHub
parent cb56f39070
commit cf9cd5da9d
2 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/autoscaler/commands.py
+14 -2
View File
@@ -210,7 +210,7 @@ def get_or_create_head_node(config, config_file, no_restart, restart_only, yes,
provider.external_ip(head_node)))
def attach_cluster(config_file, start, use_tmux, override_cluster_name):
def attach_cluster(config_file, start, use_tmux, override_cluster_name, new):
"""Attaches to a screen for the specified cluster.
Arguments:
@@ -218,8 +218,20 @@ def attach_cluster(config_file, start, use_tmux, override_cluster_name):
start: whether to start the cluster if it isn't up
use_tmux: whether to use tmux as multiplexer
override_cluster_name: set the name of the cluster
new: whether to force a new screen
"""
cmd = "tmux attach || tmux new" if use_tmux else "screen -L -xRR"
if use_tmux:
if new:
cmd = "tmux new"
else:
cmd = "tmux attach || tmux new"
else:
if new:
cmd = "screen -L"
else:
cmd = "screen -L -xRR"
exec_cluster(config_file, cmd, False, False, False, start,
override_cluster_name, None)