Quick fix for NAS to pass ut (#1212)

* fix a bug * fix a bug
microsoft · Jun 25, 2019 · 524fbc1 · 524fbc1
1 parent 8b8ff6f
commit 524fbc1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)