From 16be483af78b56eaf2dc277c71347fe2b947ae19 Mon Sep 17 00:00:00 2001 From: Philipp Moritz Date: Tue, 10 Dec 2019 10:25:01 -0800 Subject: [PATCH] [Projects] Return parameters for a command (#6409) --- python/ray/projects/scripts.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/ray/projects/scripts.py b/python/ray/projects/scripts.py index f2b67a38f..028f1ac6f 100644 --- a/python/ray/projects/scripts.py +++ b/python/ray/projects/scripts.py @@ -270,10 +270,11 @@ def get_session_runs(name, command, parsed_args): List of sessions to start, which are dictionaries with keys: "name": Name of the session to start, "command": Command to run after starting the session, + "params": Parameters for this run, "num_steps": 4 if a command should be run, 3 if not. """ if not command: - return [{"name": name, "command": None, "num_steps": 3}] + return [{"name": name, "command": None, "params": {}, "num_steps": 3}] # Try to find a wildcard argument (i.e. one that has a list of values) # and give an error if there is more than one (currently unsupported). @@ -290,6 +291,7 @@ def get_session_runs(name, command, parsed_args): session_run = { "name": name, "command": format_command(command, parsed_args), + "params": parsed_args, "num_steps": 4 } return [session_run] @@ -301,6 +303,7 @@ def get_session_runs(name, command, parsed_args): session_run = { "name": "{}-{}-{}".format(name, wildcard_arg, val), "command": format_command(command, parsed_args), + "params": parsed_args, "num_steps": 4 } session_runs.append(session_run)