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)