Fix PAI virtualCluster in non-reuse mode (#4061)

microsoft · Aug 11, 2021 · 9a4d0d6 · 9a4d0d6
1 parent f93e774
commit 9a4d0d6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ts/nni_manager/training_service/pai/paiTrainingService.ts b/ts/nni_manager/training_service/pai/paiTrainingService.ts
@@ -410,9 +410,9 @@ class PAITrainingService implements TrainingService {
                     submitFrom: 'submit-job-v2'
                 }
             }
-            if (this.config.deprecated && this.config.deprecated.virtualCluster) {
+            if (this.config.virtualCluster) {
                 nniJobConfig.defaults = {
-                    virtualCluster: this.config.deprecated.virtualCluster
+                    virtualCluster: this.config.virtualCluster
                 }
             }
         }