From 2c5daa1c1297c942f99bbfe3a5108053c3b4df82 Mon Sep 17 00:00:00 2001
From: chenyifan01 <cyf_uestc@outlook.com>
Date: Wed, 27 Jul 2022 09:56:39 +0800
Subject: [PATCH] #2588 fix bug

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

diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index 25a76fa41..43f4a6e73 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -294,6 +294,9 @@ func NotebookShow(ctx *context.Context) {
 		} else { //deleted record
 
 		}
+		if task.FlavorCode == "" {
+			task.FlavorCode = result.Flavor
+		}
 	}
 
 	datasetDownload := make([]models.DatasetDownload, 0)
@@ -528,6 +531,8 @@ func NotebookManage(ctx *context.Context) {
 				Description:     task.Description,
 				CreatedUnix:     createTime,
 				UpdatedUnix:     createTime,
+				FlavorCode:      task.FlavorCode,
+				FlavorName:      task.FlavorName,
 			}
 
 			err = models.RestartCloudbrain(task, newTask)