From e9ec5e63880fbac0e9d4e4e57ea9059c1c0927c5 Mon Sep 17 00:00:00 2001
From: liuzx <liuzx@pcl.ac.cn>
Date: Wed, 22 Jun 2022 18:08:23 +0800
Subject: [PATCH] fix-bug

---
 routers/repo/modelarts.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index b1864136a..0a4a9bbc1 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -1244,11 +1244,17 @@ func TrainJobCreateVersion(ctx *context.Context, form auth.CreateModelArtsTrainJ
 	EngineName := form.EngineName
 	isLatestVersion := modelarts.IsLatestVersion
 
+	if IsDatasetUseCountExceed(uuid) {
+		log.Error("DatasetUseCount is Exceed:%v")
+		versionErrorDataPrepare(ctx, form)
+		ctx.RenderWithErr("DatasetUseCount is Exceed", tplModelArtsTrainJobVersionNew, &form)
+		return
+	}
 	datasetName, err := GetDatasetNameByUUID(uuid)
 	if err != nil {
 		log.Error("GetDatasetNameByUUID failed:%v", err, ctx.Data["MsgID"])
-		trainJobErrorNewDataPrepare(ctx, form)
-		ctx.RenderWithErr("GetDatasetNameByUUID error", tplModelArtsTrainJobNew, &form)
+		versionErrorDataPrepare(ctx, form)
+		ctx.RenderWithErr("GetDatasetNameByUUID error", tplModelArtsTrainJobVersionNew, &form)
 		return
 	}
 	dataPath := GetObsDataPathByUUID(uuid)