Change hard-coded root directory to $PWD in PAI container (#182)

src/nni_manager/training_service/pai/hdfsClientUtility.ts

@@ -133,10 +133,13 @@ export namespace HDFSClientUtility {
              deferred.resolve(exist);
         });
 
-        // Set timeout and reject the promise once reach timeout (5 seconds)
-        setTimeout(() => deferred.reject(`Check HDFS path ${hdfsPath} exists timeout`), 5000);
+        let timeoutId : NodeJS.Timer
+        const delayTimeout : Promise<boolean> = new Promise<boolean>((resolve : Function, reject : Function) : void => {
+            // Set timeout and reject the promise once reach timeout (5 seconds)        
+            setTimeout(() => deferred.reject(`Check HDFS path ${hdfsPath} exists timeout`), 5000);
+        });
 
-        return deferred.promise;
+        return Promise.race([deferred.promise, delayTimeout]).finally(() => clearTimeout(timeoutId));
     }
 
     /**

src/nni_manager/training_service/pai/paiTrainingService.ts

@@ -186,8 +186,8 @@ class PAITrainingService implements TrainingService {
         const nniPaiTrialCommand : string = String.Format(
             PAI_TRIAL_COMMAND_FORMAT,
             // PAI will copy job's codeDir into /root directory
-            `/root/${trialJobId}`,
-            `/root/${trialJobId}/nnioutput`,
+            `$PWD/${trialJobId}`,
+            `$PWD/${trialJobId}/nnioutput`,
             trialJobId,
             this.experimentId,
             this.paiTrialConfig.command, 
@@ -343,7 +343,17 @@ class PAITrainingService implements TrainingService {
                         deferred.resolve();
                     }
                 });
-                break;
+
+                let timeoutId: NodeJS.Timer;
+                const timeoutDelay: Promise<void> = new Promise<void>((resolve: Function, reject: Function): void => {
+                    // Set timeout and reject the promise once reach timeout (5 seconds)
+                    timeoutId = setTimeout(
+                        () => reject(new Error('Get PAI token timeout. Please check your PAI cluster.')),
+                        5000);
+                });
+
+                return Promise.race([timeoutDelay, deferred.promise]).finally(() => clearTimeout(timeoutId));
+
             case TrialConfigMetadataKey.TRIAL_CONFIG:
                 if (!this.paiClusterConfig){
                     this.log.error('pai cluster config is not initialized');