From 83e987abde89c1f776584d621bda940cf5922629 Mon Sep 17 00:00:00 2001
From: chenyifan01 <cyf_uestc@outlook.com>
Date: Wed, 18 Jan 2023 14:51:43 +0800
Subject: [PATCH] #2901 fix bug

---
 routers/repo/cloudbrain.go | 13 +++++--------
 1 file changed, 5 insertions(+), 8 deletions(-)

diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index 6d25511cc..64075e035 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -1353,14 +1353,11 @@ 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)
-					return err
-				})
-				logErrorAndUpdateJobStatus(err, taskInfo)
-			}
-
+			err := retry(3, time.Second*30, func() error {
+				_, err := grampus.StopJob(taskInfo.JobID, taskInfo.JobType)
+				return err
+			})
+			logErrorAndUpdateJobStatus(err, taskInfo)
 		}
 	}
 }