Fix aml v2 config (#3655)

dfe3c27b · SparkSnail · GitHub · 8f002c8b · dfe3c27b
Unverified Commit dfe3c27b authored May 21, 2021 by SparkSnail Committed by GitHub May 21, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

ts/nni_manager/training_service/reusable/environments/amlEnvironmentService.ts ...ng_service/reusable/environments/amlEnvironmentService.ts +3 -1

No files found.
--- a/ts/nni_manager/training_service/reusable/environments/amlEnvironmentService.ts
+++ b/ts/nni_manager/training_service/reusable/environments/amlEnvironmentService.ts
@@ -101,7 +101,9 @@ export class AMLEnvironmentService extends EnvironmentService {
            amlEnvironment.command = `mv envs outputs/envs && cd outputs && ${amlEnvironment.command}`;
        }
        amlEnvironment.command = `import os\nos.system('${amlEnvironment.command}')`;
-        amlEnvironment.useActiveGpu = !!this.config.deprecated.useActiveGpu;
+        if (this.config.deprecated && this.config.deprecated.useActiveGpu !== undefined) {
+            amlEnvironment.useActiveGpu = this.config.deprecated.useActiveGpu;
+        }
        amlEnvironment.maxTrialNumberPerGpu = this.config.maxTrialNumberPerGpu;

        await fs.promises.writeFile(path.join(environmentLocalTempFolder, 'nni_script.py'), amlEnvironment.command, { encoding: 'utf8' });