diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index bfdcff6a4..6f5ffba00 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -957,15 +957,18 @@ func trainJobNewVersionDataPrepare(ctx *context.Context) error {
 		return err
 	}
 	_, _, datasetNames, _, err := getDatasUrlListByUUIDS(task.Uuid)
-	// if err != nil {
-	// 	ctx.ServerError("GetAllUserAttachments failed:", err)
-	// 	return err
-	// }
+	if err != nil {
+		log.Info("query dataset error," + err.Error())
+		//ctx.ServerError("GetAllUserAttachments failed:", err)
+		//return err
+	} else {
+		ctx.Data["dataset_name"] = datasetNames
+	}
 	ctx.Data["branches"] = branches
 	ctx.Data["branch_name"] = task.BranchName
 	ctx.Data["description"] = task.Description
 	ctx.Data["boot_file"] = task.BootFile
-	ctx.Data["dataset_name"] = datasetNames
+
 	ctx.Data["work_server_number"] = task.WorkServerNumber
 	ctx.Data["flavor_name"] = task.FlavorName
 	ctx.Data["engine_name"] = task.EngineName