From ce2e2ecdd4db18ce7ba1ae6f9ce4331da4b70958 Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Tue, 22 Nov 2022 15:15:08 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modules/grampus/resty.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/modules/grampus/resty.go b/modules/grampus/resty.go
index 13e6866fc..f615d96f2 100755
--- a/modules/grampus/resty.go
+++ b/modules/grampus/resty.go
@@ -257,11 +257,16 @@ func GetImages(processorType string, jobType string) (*models.GetGrampusImagesRe
 
 	retry := 0
 
+	queryType := "TrainJob"
+	if jobType == string(models.JobTypeDebug) {
+		queryType = "Notebook"
+	}
+
 sendjob:
 	_, err := client.R().
 		SetAuthToken(TOKEN).
 		SetResult(&result).
-		Get(HOST + urlGetImages + "?processorType=" + processorType + "&jobType=" + jobType)
+		Get(HOST + urlGetImages + "?processorType=" + processorType + "&trainType=" + queryType)
 
 	if err != nil {
 		return nil, fmt.Errorf("resty GetImages: %v", err)