fix support setting nniManagerIp in PAI (#1987)

c55c5f46 · SparkSnail · GitHub · 5dde0d8d · c55c5f46
Unverified Commit c55c5f46 authored Feb 03, 2020 by SparkSnail Committed by GitHub Feb 03, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

src/nni_manager/training_service/pai/paiK8S/paiK8STrainingService.ts ...ager/training_service/pai/paiK8S/paiK8STrainingService.ts +4 -0

No files found.
--- a/src/nni_manager/training_service/pai/paiK8S/paiK8STrainingService.ts
+++ b/src/nni_manager/training_service/pai/paiK8S/paiK8STrainingService.ts
@@ -59,6 +59,10 @@ class PAIK8STrainingService extends PAITrainingService {

    public async setClusterMetadata(key: string, value: string): Promise<void> {
        switch (key) {
+            case TrialConfigMetadataKey.NNI_MANAGER_IP:
+                this.nniManagerIpConfig = <NNIManagerIpConfig>JSON.parse(value);
+                break;
+
            case TrialConfigMetadataKey.PAI_CLUSTER_CONFIG:
                this.paiJobRestServer = new PAIJobRestServer(component.get(PAIK8STrainingService));
                this.paiClusterConfig = <PAIClusterConfig>JSON.parse(value);