From e197296f1ae7141b15ec5a114471e6226b25f51d Mon Sep 17 00:00:00 2001
From: yuyuanshifu <747342561@qq.com>
Date: Thu, 18 Mar 2021 17:31:47 +0800
Subject: [PATCH] mod taskinfo

---
 models/cloudbrain.go       | 2 +-
 routers/repo/cloudbrain.go | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/models/cloudbrain.go b/models/cloudbrain.go
index d9db9ef23..b97428861 100755
--- a/models/cloudbrain.go
+++ b/models/cloudbrain.go
@@ -165,7 +165,7 @@ type TaskInfo struct {
 	Username string `json:"username"`
 	TaskName string `json:"task_name"`
 	CodeName string `json:"code_name"`
-	BenchmarkCategory string `json:"selected_category"`
+	BenchmarkCategory []string `json:"selected_category"`
 }
 
 func ConvertToTaskPod(input map[string]interface{}) (TaskPod, error) {
diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index 952177d61..be8e5d14e 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -368,7 +368,7 @@ func downloadRateCode(repo *models.Repository, taskName, gitPath, codePath, benc
 		Username: repo.Owner.Name,
 		TaskName: taskName,
 		CodeName: repo.Name,
-		BenchmarkCategory: benchmarkCategory,
+		BenchmarkCategory: strings.Split(benchmarkCategory, ","),
 	})
 	if err != nil {
 		log.Error("json.Marshal failed", err.Error())