Quick fix for NAS to pass ut (#1212)

* fix a bug * fix a bug

Quick fix for NAS to pass ut (#1212)
* fix a bug * fix a bug
524fbc1c · Zejun Lin · Yan Ni · 8b8ff6f7 · 524fbc1c
Commit 524fbc1c authored Jun 25, 2019 by Zejun Lin Committed by Yan Ni Jun 25, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tools/nni_cmd/launcher.py tools/nni_cmd/launcher.py +2 -2

No files found.
--- a/tools/nni_cmd/launcher.py
+++ b/tools/nni_cmd/launcher.py
@@ -377,8 +377,8 @@ def launch_experiment(args, experiment_config, mode, config_file_name, experimen
        if not os.path.isdir(path):
            os.makedirs(path)
        path = tempfile.mkdtemp(dir=path)
-        nas_mode = experiment_config['trial'].get('nasMode')
-        code_dir = expand_annotations(experiment_config['trial']['codeDir'], path, nas_mode)
+        nas_mode = experiment_config['trial'].get('nasMode', 'classic_mode')
+        code_dir = expand_annotations(experiment_config['trial']['codeDir'], path, nas_mode=nas_mode)
        experiment_config['trial']['codeDir'] = code_dir
        search_space = generate_search_space(code_dir)
        experiment_config['searchSpace'] = json.dumps(search_space)