From 4ad1e8957e04810bb9c4dc06ae0e865893fdae20 Mon Sep 17 00:00:00 2001 From: zouap Date: Mon, 10 Oct 2022 14:32:23 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=B1=9E=E6=80=A7=E7=BB=99?= =?UTF-8?q?=E4=BA=88?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: zouap --- routers/repo/aisafety.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go index 58baf1a0c..091ae022f 100644 --- a/routers/repo/aisafety.go +++ b/routers/repo/aisafety.go @@ -95,6 +95,9 @@ func GetAiSafetyTask(ctx *context.Context) { job, err = models.GetCloudbrainByIDWithDeleted(ID) job.BenchmarkType = "安全评测" job.BenchmarkTypeName = "Image Classification" + job.CanModify = cloudbrain.CanModifyJob(ctx, job) + job.CanDel = cloudbrain.CanDeleteJob(ctx, job) + ctx.JSON(200, job) }