diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go index a820c56fa..484715cb5 100755 --- a/routers/repo/modelarts.go +++ b/routers/repo/modelarts.go @@ -1160,6 +1160,14 @@ func TrainJobShow(ctx *context.Context) { ctx.RenderWithErr(err.Error(), tplModelArtsTrainJobShow, nil) return } + //设置权限 + canNewJob, err := canUserCreateTrainJobVersion(ctx, VersionListTasks[0].UserID) + if err != nil { + ctx.ServerError("canNewJob failed", err) + return + } + ctx.Data["canNewJob"] = canNewJob + //将运行参数转化为epoch_size = 3, device_target = Ascend的格式 for i, _ := range VersionListTasks { @@ -1311,19 +1319,14 @@ func canUserCreateTrainJob(uid int64) (bool, error) { return org.IsOrgMember(uid) } -func canUserCreateTrainJobVersion(ctx *context.Context, jobID string, versionName string) (bool, error) { - // task, err := models.GetCloudbrainByJobIDAndVersionName(jobID, versionName) - // if err != nil { - // return false, err - // } - // if ctx.User.ID == task.UserID { - // canNewJob := true - // return canNewJob, nil - // } else { - // canNewJob := false - // return canNewJob, nil - // } - return true, nil +func canUserCreateTrainJobVersion(ctx *context.Context, userID int64) (bool, error) { + if ctx.User.ID == userID || ctx.User.IsAdmin { + canNewJob := true + return canNewJob, nil + } else { + canNewJob := false + return canNewJob, nil + } } func TrainJobGetConfigList(ctx *context.Context) {