From 459fd5e152cd07b17538800cb2bbd779c54fcfd4 Mon Sep 17 00:00:00 2001 From: Richard Liaw Date: Thu, 15 Mar 2018 18:41:02 -0700 Subject: [PATCH] [tune][minor] Move helper function (#1722) --- python/ray/tune/variant_generator.py | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/python/ray/tune/variant_generator.py b/python/ray/tune/variant_generator.py index 4224073c3..b12b6ca68 100644 --- a/python/ray/tune/variant_generator.py +++ b/python/ray/tune/variant_generator.py @@ -10,6 +10,17 @@ from ray.tune.trial import Trial from ray.tune.config_parser import make_parser, json_to_resources +def to_argv(config): + argv = [] + for k, v in config.items(): + argv.append("--{}".format(k.replace("_", "-"))) + if isinstance(v, str): + argv.append(v) + else: + argv.append(json.dumps(v)) + return argv + + def generate_trials(unresolved_spec, output_path=''): """Wraps `generate_variants()` to return a Trial object for each variant. @@ -23,17 +34,6 @@ def generate_trials(unresolved_spec, output_path=''): if "run" not in unresolved_spec: raise TuneError("Must specify `run` in {}".format(unresolved_spec)) - - def to_argv(config): - argv = [] - for k, v in config.items(): - argv.append("--{}".format(k.replace("_", "-"))) - if isinstance(v, str): - argv.append(v) - else: - argv.append(json.dumps(v)) - return argv - parser = make_parser() i = 0 for _ in range(unresolved_spec.get("repeat", 1)):