catch error

microsoft · Dec 10, 2020 · 1325371 · 1325371
1 parent 956a0f0
commit 1325371
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/ts/nni_manager/training_service/local/localTrainingService.ts b/ts/nni_manager/training_service/local/localTrainingService.ts
@@ -256,11 +256,11 @@ class LocalTrainingService implements TrainingService {
         }
         tkill(trialJob.pid, 'SIGTERM');
         const pid = trialJob.pid;
-        setTimeout((pid: number) => {
-            if (isAlive(pid)) {
-                tkill(pid, 'SIGKILL');
-            }
-        }, 5 * 1000, pid);
+        setTimeout(((pid: number) => {
+            tkill(pid, 'SIGKILL', (err) => {
+                this.log.warning(`cancel trial job {pid: ${pid}} failed: ${err?.message}`);
+            });
+        }).bind(this), 5 * 1000, pid);
 
         this.setTrialJobStatus(trialJob, getJobCancelStatus(isEarlyStopped));