From 6dc738b82230101f4eef7c17c54e6318a773ea0f Mon Sep 17 00:00:00 2001
From: liuzx <liuzx@pcl.ac.cn>
Date: Thu, 11 Nov 2021 15:40:54 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=95=B0=E6=8D=AE=E5=BA=93?=
 =?UTF-8?q?=E7=9A=84=E5=90=AF=E5=8A=A8=E6=96=87=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modules/modelarts/modelarts.go | 6 ++++--
 routers/repo/modelarts.go      | 9 ++++-----
 2 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/modules/modelarts/modelarts.go b/modules/modelarts/modelarts.go
index 39601a600..8db1c11bf 100755
--- a/modules/modelarts/modelarts.go
+++ b/modules/modelarts/modelarts.go
@@ -66,6 +66,7 @@ type GenerateTrainJobReq struct {
 	Description       string
 	CodeObsPath       string
 	BootFile          string
+	BootFileUrl       string
 	DataUrl           string
 	TrainUrl          string
 	FlavorCode        string
@@ -89,6 +90,7 @@ type GenerateTrainJobVersionReq struct {
 	Description      string
 	CodeObsPath      string
 	BootFile         string
+	BootFileUrl      string
 	DataUrl          string
 	TrainUrl         string
 	FlavorCode       string
@@ -210,7 +212,7 @@ func GenerateTrainJob(ctx *context.Context, req *GenerateTrainJobReq) (err error
 		Config: models.Config{
 			WorkServerNum: req.WorkServerNumber,
 			AppUrl:        req.CodeObsPath,
-			BootFileUrl:   req.BootFile,
+			BootFileUrl:   req.BootFileUrl,
 			DataUrl:       req.DataUrl,
 			EngineID:      req.EngineID,
 			TrainUrl:      req.TrainUrl,
@@ -278,7 +280,7 @@ func GenerateTrainJobVersion(ctx *context.Context, req *GenerateTrainJobVersionR
 		Config: models.TrainJobVersionConfig{
 			WorkServerNum: req.WorkServerNumber,
 			AppUrl:        req.CodeObsPath,
-			BootFileUrl:   req.BootFile,
+			BootFileUrl:   req.BootFileUrl,
 			DataUrl:       req.DataUrl,
 			EngineID:      req.EngineID,
 			TrainUrl:      req.TrainUrl,
diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index bf4a1f31f..5ca28515e 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -848,7 +848,8 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)
 		DataUrl:           dataPath,
 		Description:       description,
 		CodeObsPath:       codeObsPath,
-		BootFile:          codeObsPath + bootFile,
+		BootFileUrl:       codeObsPath + bootFile,
+		BootFile:          bootFile,
 		TrainUrl:          outputObsPath,
 		FlavorCode:        flavorCode,
 		WorkServerNumber:  workServerNumber,
@@ -912,9 +913,6 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)
 func TrainJobCreateVersion(ctx *context.Context, form auth.CreateModelArtsTrainJobForm) {
 	ctx.Data["PageIsTrainJob"] = true
 	var jobID = ctx.Params(":jobid")
-	// var fatherVersionName = ctx.Query("versionName")
-	// jobID = "19373"
-	// versionName = "V0009"
 
 	jobName := form.JobName
 	uuid := form.Attachment
@@ -1072,7 +1070,8 @@ func TrainJobCreateVersion(ctx *context.Context, form auth.CreateModelArtsTrainJ
 		DataUrl:          dataPath,
 		Description:      description,
 		CodeObsPath:      codeObsPath,
-		BootFile:         codeObsPath + bootFile,
+		BootFileUrl:      codeObsPath + bootFile,
+		BootFile:         bootFile,
 		TrainUrl:         outputObsPath,
 		FlavorCode:       flavorCode,
 		WorkServerNumber: workServerNumber,