获取引擎名称

routers/repo/modelarts.go

@@ -842,6 +842,13 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)
 			return
 		}
 	}
+	//将引擎id转化为引擎名称
+	FlavorName, err = getFlavorNameByEngineID(engineID)
+	if err != nil {
+		log.Error("getFlavorNameByEngineID(%s) failed:%v", engineID, err.Error())
+		ctx.RenderWithErr(err.Error(), tplModelArtsTrainJobNew, &form)
+		return
+	}
 
 	req := &modelarts.GenerateTrainJobReq{
 		JobName:           jobName,
@@ -1057,8 +1064,14 @@ func TrainJobCreateVersion(ctx *context.Context, form auth.CreateModelArtsTrainJ
 			return
 		}
 	}
-	// JobVersionName := "V0001"
-	// PreVersionId := int64(67646)
+	//将引擎id转化为引擎名称
+	FlavorName, err = getFlavorNameByEngineID(engineID)
+	if err != nil {
+		log.Error("getFlavorNameByEngineID(%s) failed:%v", engineID, err.Error())
+		ctx.RenderWithErr(err.Error(), tplModelArtsTrainJobVersionNew, &form)
+		return
+	}
+
 	task, err := models.GetCloudbrainByJobIDAndVersionName(jobID, fatherVersionName)
 	if err != nil {
 		log.Error("GetCloudbrainByJobIDAndVersionName(%s) failed:%v", jobID, err.Error())
@@ -1128,6 +1141,19 @@ func TrainJobCreateVersion(ctx *context.Context, form auth.CreateModelArtsTrainJ
 	ctx.HTML(http.StatusOK, tplModelArtsTrainJobShow)
 }
 
+func getFlavorNameByEngineID(engineID int) (FlavorName string, err error) {
+	if engineID == 121 {
+		FlavorName = "TensorFlow-1.15"
+		return FlavorName, nil
+	} else if engineID == 122 {
+		FlavorName = "Mindspore-1.3"
+		return FlavorName, nil
+	} else {
+		log.Error("getFlavorName failed:%v", errors.New("getFlavorName failed"))
+		return "getFlavorName failed:", errors.New("getFlavorName failed")
+	}
+}
+
 // readDir reads the directory named by dirname and returns
 // a list of directory entries sorted by filename.
 func readDir(dirname string) ([]os.FileInfo, error) {