diff --git a/advisor/internal/scheduler/scheduler.go b/advisor/internal/scheduler/scheduler.go index 672dde6..b758961 100644 --- a/advisor/internal/scheduler/scheduler.go +++ b/advisor/internal/scheduler/scheduler.go @@ -362,15 +362,17 @@ func (s *DefaultSchedule) ComputeAllSlwNodeScore(files jobmod.JobFiles) (map[int for id, image := range imageStgScores { // TODO 此处id错误,根据算力中心id做判断,待修改 var isLoaded bool - if id == res.SlwNodeImageID { - imageResp, err := colCli.GetImageList(collector.NewGetImageList(id)) - if err != nil { - return nil, err - } - for _, imageID := range imageResp.ImageIDs { - if imageID == res.SlwNodeImageID { - isLoaded = true - break + for _, info := range res.ImportingInfos { + if id == info.SlwNodeID { + imageResp, err := colCli.GetImageList(collector.NewGetImageList(id)) + if err != nil { + return nil, err + } + for _, imageID := range imageResp.ImageIDs { + if imageID == info.SlwNodeImageID { + isLoaded = true + break + } } } }