From 9b292bf927ccb2af0b9de7067f96dab3e029bd11 Mon Sep 17 00:00:00 2001
From: lewis <747342561@qq.com>
Date: Tue, 18 Jan 2022 15:24:43 +0800
Subject: [PATCH] train_url

---
 modules/auth/cloudbrain.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/auth/cloudbrain.go b/modules/auth/cloudbrain.go
index 277588f40..cfa1e81c3 100755
--- a/modules/auth/cloudbrain.go
+++ b/modules/auth/cloudbrain.go
@@ -13,6 +13,8 @@ type CreateCloudBrainForm struct {
 	JobType              string `form:"job_type" binding:"Required"`
 	BenchmarkCategory    string `form:"get_benchmark_category"`
 	GpuType              string `form:"gpu_type"`
+	TrainUrl             string `form:"train_url"`
+	TestUrl              string `form:"test_url"`
 	ResourceSpecId       int    `form:"resource_spec_id" binding:"Required"`
 	BenchmarkTypeID      int    `form:"benchmark_types_id"`
 	BenchmarkChildTypeID int    `form:"benchmark_child_types_id"`