From 894796ec87f0646e73dcb1712cca1cda1ca6a2fb Mon Sep 17 00:00:00 2001
From: songjc <969378911@qq.com>
Date: Tue, 19 Sep 2023 15:25:03 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=20=E8=B0=83=E5=BA=A6?=
 =?UTF-8?q?=E6=96=B9=E6=A1=88?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 advisor/internal/scheduler/scheduler.go | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

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
+					}
 				}
 			}
 		}