From f633bf4d57d9c03a91f8a435bbf69e21ded1f1b6 Mon Sep 17 00:00:00 2001
From: lewis <747342561@qq.com>
Date: Thu, 25 Nov 2021 18:22:16 +0800
Subject: [PATCH] add branch name

---
 routers/repo/modelarts.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index 9ea5de5c3..069a1a0b5 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -370,6 +370,7 @@ func trainJobNewDataPrepare(ctx *context.Context) error {
 	ctx.Data["Branches"] = Branches
 	ctx.Data["BranchesCount"] = len(Branches)
 	ctx.Data["params"] = ""
+	ctx.Data["BranchName"] = ctx.Repo.BranchName
 
 	configList, err := getConfigList(modelarts.PerPage, 1, modelarts.SortByCreateTime, "desc", "", modelarts.ConfigTypeCustom)
 	if err != nil {