From dadc4b98d452f6a6a00e9f7ca96f59cad8fa8d9f Mon Sep 17 00:00:00 2001
From: lewis <747342561@qq.com>
Date: Tue, 29 Mar 2022 15:44:33 +0800
Subject: [PATCH] notebook opt

---
 routers/repo/modelarts.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index 02a46b492..a9188afd4 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -410,8 +410,7 @@ func NotebookManage(ctx *context.Context) {
 			break
 		}
 
-		task.Status = res.Status
-		status = task.Status
+		status = res.Status
 		if action == models.ActionStart {
 			newTask := &models.Cloudbrain{
 				Status:          status,
@@ -435,7 +434,9 @@ func NotebookManage(ctx *context.Context) {
 				errorMsg = "system error"
 				break
 			}
+			ID = strconv.FormatInt(newTask.ID, 10)
 		} else {
+			task.Status = res.Status
 			if task.EndTime == 0 && models.IsModelArtsDebugJobTerminal(task.Status) {
 				task.EndTime = timeutil.TimeStampNow()
 			}