From f270b7b38c797796b0e01b70a57076e7509b1ef4 Mon Sep 17 00:00:00 2001 From: liuzx Date: Mon, 6 Dec 2021 18:52:41 +0800 Subject: [PATCH] fixbug-1022 --- routers/repo/modelarts.go | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go index 8b09164d3..a2a20df78 100755 --- a/routers/repo/modelarts.go +++ b/routers/repo/modelarts.go @@ -67,19 +67,20 @@ func DebugJobIndex(ctx *context.Context) { for i, task := range ciTasks { if task.Cloudbrain.Type == models.TypeCloudBrainOne { if task.Status == string(models.JobRunning) && (timestamp-int64(task.Cloudbrain.CreatedUnix) > 10) { - ciTasks[i].CanDebug = true + ciTasks[i].CanDebug = cloudbrain.CanCreateOrDebugJob(ctx) } else { ciTasks[i].CanDebug = false } - ciTasks[i].CanDel = models.CanDelJob(ctx.IsSigned, ctx.User, task) + ciTasks[i].CanDel = cloudbrain.CanDeleteDebugJob(ctx, &task.Cloudbrain) ciTasks[i].Cloudbrain.ComputeResource = modelarts.GPUResource } if task.Cloudbrain.Type == models.TypeCloudBrainTwo { if task.Status == string(models.JobRunning) { - ciTasks[i].CanDebug = true + ciTasks[i].CanDebug = cloudbrain.CanCreateOrDebugJob(ctx) } else { ciTasks[i].CanDebug = false } + ciTasks[i].CanDel = cloudbrain.CanDeleteDebugJob(ctx, &task.Cloudbrain) ciTasks[i].Cloudbrain.ComputeResource = modelarts.NPUResource } @@ -88,9 +89,9 @@ func DebugJobIndex(ctx *context.Context) { pager := context.NewPagination(int(count), setting.UI.IssuePagingNum, page, 5) pager.SetDefaultParams(ctx) ctx.Data["Page"] = pager - ctx.Data["PageIsCloudBrain"] = true ctx.Data["Tasks"] = ciTasks + ctx.Data["CanCreate"] = cloudbrain.CanCreateOrDebugJob(ctx) ctx.HTML(200, tplDebugJobIndex) }