Fix local IT

nni/experiment/experiment.py

@@ -169,6 +169,7 @@ def _stop_nni_manager(self) -> None:
         if self._proc is not None:
             try:
                 rest.delete(self.port, '/experiment', self.url_prefix)
+                self._proc.wait()
             except Exception as e:
                 _logger.exception(e)
                 _logger.warning('Cannot gracefully stop experiment, killing NNI process...')

ts/nni_manager/common/trial_keeper/keeper.ts

@@ -89,12 +89,13 @@ export class TrialKeeper {
 
     // TODO: support user configurable init command
     public async start(): Promise<EnvironmentInfo> {
+        this.envInfo = { id: this.envId, type: 'hot' } as EnvironmentInfo;
+
         await Promise.all([
             this.scheduler.start(),
             this.channels.start(),
         ]);
 
-        this.envInfo = { id: this.envId, type: 'hot' } as EnvironmentInfo;
         Object.assign(this.envInfo, await collectPlatformInfo(this.gpuEnabled));
         return this.envInfo;
     }