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

[autoscaler] Fix submit (#4782)

Browse Source
This commit is contained in:
Richard Liaw
2019-05-14 19:52:28 -07:00
committed by GitHub
parent c5161a2c4d
commit 3bbafc7105
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/scripts/scripts.py
+7 -4
View File
@@ -555,7 +555,7 @@ def rsync_up(cluster_config_file, source, target, cluster_name):
rsync(cluster_config_file, source, target, cluster_name, down=False)
@cli.command()
@cli.command(context_settings={"ignore_unknown_options": True})
@click.argument("cluster_config_file", required=True, type=str)
@click.option(
"--docker",
@@ -588,14 +588,17 @@ def rsync_up(cluster_config_file, source, target, cluster_name):
@click.option(
"--port-forward", required=False, type=int, help="Port to forward.")
@click.argument("script", required=True, type=str)
@click.argument("script_args", required=False, type=str, nargs=-1)
@click.option("--args", required=False, type=str, help="Script args.")
def submit(cluster_config_file, docker, screen, tmux, stop, start,
cluster_name, port_forward, script, script_args):
cluster_name, port_forward, script, args):
"""Uploads and runs a script on the specified cluster.
The script is automatically synced to the following location:
os.path.join("~", os.path.basename(script))
Example:
>>> ray submit [CLUSTER.YAML] experiment.py --args="--smoke-test"
"""
assert not (screen and tmux), "Can specify only one of `screen` or `tmux`."
@@ -606,7 +609,7 @@ def submit(cluster_config_file, docker, screen, tmux, stop, start,
target = os.path.join("~", os.path.basename(script))
rsync(cluster_config_file, script, target, cluster_name, down=False)
cmd = " ".join(["python", target] + list(script_args))
cmd = " ".join(["python", target, args])
exec_cluster(cluster_config_file, cmd, docker, screen, tmux, stop, False,
cluster_name, port_forward)