Fix aml cluster metadata empty bug (#3015)

6aae16c5 · SparkSnail · GitHub · 60ed8c3b · 6aae16c5
Unverified Commit 6aae16c5 authored Oct 21, 2020 by SparkSnail Committed by GitHub Oct 21, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

tools/nni_cmd/launcher.py tools/nni_cmd/launcher.py +5 -0

No files found.
--- a/tools/nni_cmd/launcher.py
+++ b/tools/nni_cmd/launcher.py
@@ -371,6 +371,11 @@ def set_experiment(experiment_config, mode, port, config_file_name):
            {'key': 'frameworkcontroller_config', 'value': experiment_config['frameworkcontrollerConfig']})
        request_data['clusterMetaData'].append(
            {'key': 'trial_config', 'value': experiment_config['trial']})
+    elif experiment_config['trainingServicePlatform'] == 'aml':
+        request_data['clusterMetaData'].append(
+            {'key': 'aml_config', 'value': experiment_config['amlConfig']})
+        request_data['clusterMetaData'].append(
+            {'key': 'trial_config', 'value': experiment_config['trial']})
    response = rest_post(experiment_url(port), json.dumps(request_data), REST_TIME_OUT, show_error=True)
    if check_response(response):
        return response