提交代码

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)