提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>

modules/modelarts/modelarts.go

@@ -772,7 +772,10 @@ func GenerateInferenceJob(ctx *context.Context, req *GenerateInferenceJobReq) (e
 		return err
 	}
 	if req.JobType == string(models.JobTypeModelSafety) {
-		notification.NotifyOtherTask(ctx.User, ctx.Repo.Repository, jobID, req.DisplayJobName, models.ActionCreateBenchMarkTask)
+		task, err := models.GetCloudbrainByJobID(jobID)
+		if err == nil {
+			notification.NotifyOtherTask(ctx.User, ctx.Repo.Repository, fmt.Sprint(task.ID), req.DisplayJobName, models.ActionCreateBenchMarkTask)
+		}
 	} else {
 		notification.NotifyOtherTask(ctx.User, ctx.Repo.Repository, jobID, req.DisplayJobName, models.ActionCreateInferenceTask)
 	}

routers/repo/cloudbrain.go

@@ -742,9 +742,6 @@ func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.Jo
 		task, err = models.GetCloudbrainByJobID(ctx.Params(":jobid"))
 	} else {
 		task, err = models.GetCloudbrainByIDWithDeleted(ctx.Params(":id"))
-		if task == nil || err != nil {
-			task, err = models.GetCloudbrainByJobID(ctx.Params(":id"))
-		}
 	}
 	if task.JobType == string(models.JobTypeModelSafety) {
 		GetAiSafetyTaskTmpl(ctx)