From cd424555f9e14dd4ba314ea2bf4957f8f5b2247f Mon Sep 17 00:00:00 2001
From: lewis <747342561@qq.com>
Date: Tue, 25 May 2021 15:56:09 +0800
Subject: [PATCH] add type info into task info

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

diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index e6e56d2c8..8d8e97377 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -26,7 +26,8 @@ const (
 )
 
 var (
-	gpuInfos *models.GpuInfos
+	gpuInfos   *models.GpuInfos
+	categories *models.Categories
 )
 
 // MustEnableDataset check if repository enable internal cb
@@ -135,8 +136,9 @@ func CloudBrainNew(ctx *context.Context) {
 	ctx.Data["benchmark_path"] = cloudbrain.BenchMarkMountPath
 	ctx.Data["is_benchmark_enabled"] = setting.IsBenchmarkEnabled
 
-	var categories *models.Categories
-	json.Unmarshal([]byte(setting.BenchmarkCategory), &categories)
+	if categories == nil {
+		json.Unmarshal([]byte(setting.BenchmarkCategory), &categories)
+	}
 	ctx.Data["benchmark_categories"] = categories.Category
 
 	if gpuInfos == nil {