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

Fix overriding of params dictionary (#6445)

Browse Source
This commit is contained in:
Philipp Moritz
2019-12-12 19:15:13 -08:00
committed by GitHub
parent 5a5c94939f
commit 74b454c614
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/projects/projects.py
+7 -5
View File
@@ -92,20 +92,22 @@ class ProjectDefinition:
# with them, save it in the following dictionary.
choices = {}
for param in params:
name = param.pop("name")
# Construct arguments to pass into argparse's parser.add_argument.
argparse_kwargs = copy.deepcopy(param)
name = argparse_kwargs.pop("name")
if wildcards and "choices" in param:
choices[name] = copy.deepcopy(param["choices"])
param["choices"] = param["choices"] + ["*"]
choices[name] = param["choices"]
argparse_kwargs["choices"] = param["choices"] + ["*"]
if "type" in param:
types = {"int": int, "str": str, "float": float}
if param["type"] in types:
param["type"] = types[param["type"]]
argparse_kwargs["type"] = types[param["type"]]
else:
raise ValueError(
"Parameter {} has type {} which is not supported. "
"Type must be one of {}".format(
name, param["type"], list(types.keys())))
parser.add_argument("--" + name, dest=name, **param)
parser.add_argument("--" + name, dest=name, **argparse_kwargs)
parsed_args = vars(parser.parse_args(list(args)))