From 02dcd1e0bbbfb142d9f9867b9e447c7222764ea4 Mon Sep 17 00:00:00 2001 From: chenyifan01 Date: Tue, 23 Jan 2024 11:17:24 +0800 Subject: [PATCH] #4651 fix bug --- .../ai_task_service/task/grampus_online_infer_task.go | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/services/ai_task_service/task/grampus_online_infer_task.go b/services/ai_task_service/task/grampus_online_infer_task.go index cb9636bdde..8c15267694 100644 --- a/services/ai_task_service/task/grampus_online_infer_task.go +++ b/services/ai_task_service/task/grampus_online_infer_task.go @@ -160,3 +160,12 @@ func (g GrampusOnlineInferTaskTemplate) LoadSpec(ctx *context.CreationContext) * ctx.Spec = spec return nil } + +func (GrampusOnlineInferTaskTemplate) GetAvailableQueues(ctx *context.CreationContext) *response.BizError { + ctx.Queues = ctx.Spec.GetAvailableQueues(models.GetAvailableCenterIdOpts{ + UserId: ctx.User.ID, + JobType: models.JobTypeDebug, + HasInternet: ctx.Request.HasInternet, + }) + return nil +} -- 2.34.1