fix-bug

routers/repo/cloudbrain.go

@@ -324,6 +324,12 @@ func cloudBrainTrainJobErrorPrepare(ctx *context.Context, form auth.CreateCloudB
 	ctx.Data["params"] = Parameters.Parameter
 	ctx.Data["boot_file"] = form.BootFile
 	ctx.Data["attachment"] = form.Attachment
+	_, datasetNames, err := models.GetDatasetInfo(form.Attachment)
+	if err != nil {
+		log.Error("GetDatasetInfo failed: %v", err, ctx.Data["MsgID"])
+		return nil
+	}
+	ctx.Data["dataset_name"] = datasetNames
 	ctx.Data["branch_name"] = form.BranchName
 	ctx.Data["datasetType"] = models.TypeCloudBrainOne
 
@@ -439,6 +445,12 @@ func cloudBrainInferenceJobErrorPrepare(ctx *context.Context, form auth.CreateCl
 	ctx.Data["params"] = Parameters.Parameter
 	ctx.Data["boot_file"] = form.BootFile
 	ctx.Data["attachment"] = form.Attachment
+	_, datasetNames, err := models.GetDatasetInfo(form.Attachment)
+	if err != nil {
+		log.Error("GetDatasetInfo failed: %v", err, ctx.Data["MsgID"])
+		return nil
+	}
+	ctx.Data["dataset_name"] = datasetNames
 	ctx.Data["branch_name"] = form.BranchName
 	ctx.Data["datasetType"] = models.TypeCloudBrainOne
 

routers/repo/grampus.go

@@ -204,6 +204,12 @@ func grampusTrainJobErrorPrepare(ctx *context.Context, processType string, form
 	ctx.Data["params"] = Parameters.Parameter
 	ctx.Data["boot_file"] = form.BootFile
 	ctx.Data["attachment"] = form.Attachment
+	_, datasetNames, err := models.GetDatasetInfo(form.Attachment)
+	if err != nil {
+		log.Error("GetDatasetInfo failed: %v", err, ctx.Data["MsgID"])
+		return nil
+	}
+	ctx.Data["dataset_name"] = datasetNames
 	ctx.Data["branch_name"] = form.BranchName
 	ctx.Data["image_id"] = form.ImageID