fix a bug (#1211)

fix bug for PR #1201
microsoft · Jun 25, 2019 · 8b8ff6f · 8b8ff6f
1 parent a587648
commit 8b8ff6f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tools/nni_cmd/launcher.py b/tools/nni_cmd/launcher.py
@@ -377,7 +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)
-        code_dir = expand_annotations(experiment_config['trial']['codeDir'], path, nas_mode=experiment_config['trial']['nasMode'])
+        nas_mode = experiment_config['trial'].get('nasMode')
+        code_dir = expand_annotations(experiment_config['trial']['codeDir'], path, nas_mode)
         experiment_config['trial']['codeDir'] = code_dir
         search_space = generate_search_space(code_dir)
         experiment_config['searchSpace'] = json.dumps(search_space)