routers/repo/cloudbrain.go

@@ -1353,15 +1353,12 @@ func StopJobs(cloudBrains []*models.Cloudbrain) {
 				logErrorAndUpdateJobStatus(err, taskInfo)
 			}
 		} else if taskInfo.Type == models.TypeC2Net {
-			if taskInfo.JobType == string(models.JobTypeTrain) {
 			err := retry(3, time.Second*30, func() error {
-					_, err := grampus.StopJob(taskInfo.JobID)
+				_, err := grampus.StopJob(taskInfo.JobID, taskInfo.JobType)
 				return err
 			})
 			logErrorAndUpdateJobStatus(err, taskInfo)
 		}
-
-		}
 	}
 }