From f6577583eab0cf24bddcc5842f96077cd3cf5629 Mon Sep 17 00:00:00 2001 From: chenyifan01 Date: Fri, 26 Jan 2024 09:19:58 +0800 Subject: [PATCH] #4547 fix bug --- services/ai_task_service/task/task_base.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/services/ai_task_service/task/task_base.go b/services/ai_task_service/task/task_base.go index 0bcb19586f..6d5cf9b3fa 100644 --- a/services/ai_task_service/task/task_base.go +++ b/services/ai_task_service/task/task_base.go @@ -289,7 +289,7 @@ func (g DefaultAITaskTemplate) Update(cloudbrainId int64) *response.BizError { //二是处于PREPARING的时间超过了配置的等待时间，此时意味着异步创建任务时间过长或者出现了未知异常 if cloudbrain.NeedActiveStop() { log.Info("AI task should active stop.cloudbrainId=%d", cloudbrainId) - if g.JobType == models.JobTypeDebug { + if g.JobType == models.JobTypeDebug || g.JobType == models.JobTypeOnlineInference { err = StopAITaskByJobNameFromRemote(cloudbrain, c.QueryNoteBookByJobName, c.StopNoteBook) } else { err = StopAITaskByJobNameFromRemote(cloudbrain, c.QueryTrainJobByJobName, c.StopTrainJob) @@ -307,7 +307,7 @@ func (g DefaultAITaskTemplate) Update(cloudbrainId int64) *response.BizError { log.Info("AI task is preparing.No need to update from remote.cloudbrainId=%d", cloudbrainId) return nil } - if g.JobType == models.JobTypeDebug { + if g.JobType == models.JobTypeDebug || g.JobType == models.JobTypeOnlineInference { err = UpdateAITaskFromRemote(cloudbrain, c.QueryNoteBook) } else { err = UpdateAITaskFromRemote(cloudbrain, c.QueryTrainJob) @@ -435,7 +435,7 @@ func (g DefaultAITaskTemplate) GetOperationProfile(cloudbrainId int64) (*entity. } var s *entity.OperationProfile var err error - if g.JobType == models.JobTypeDebug { + if g.JobType == models.JobTypeDebug || g.JobType == models.JobTypeOnlineInference { s, err = GetOperationProfile(cloudbrainId, c.GetNoteBookOperationProfile) } else { s, err = GetOperationProfile(cloudbrainId, c.GetTrainJobOperationProfile) -- 2.34.1