提交代码

modules/grampus/grampus.go

@@ -121,6 +121,21 @@ func getDatasetGrampus(datasetInfos map[string]models.DatasetInfo) []models.Gram
 	}
 	return datasetGrampus
 }
+func getDatasetGPUGrampus(datasetInfos map[string]models.DatasetInfo) []models.GrampusDataset {
+	var datasetGrampus []models.GrampusDataset
+	for _, datasetInfo := range datasetInfos {
+		datasetGrampus = append(datasetGrampus, models.GrampusDataset{
+			Name:          datasetInfo.FullName,
+			Bucket:        setting.Attachment.Minio.Bucket,
+			EndPoint:      setting.Attachment.Minio.Endpoint,
+			ObjectKey:     datasetInfo.DataLocalPath + "/" + datasetInfo.FullName,
+			ReadOnly:      true,
+			ContainerPath: "/dataset/" + datasetInfo.Name,
+		})
+
+	}
+	return datasetGrampus
+}
 
 func GenerateNotebookJob(ctx *context.Context, req *GenerateNotebookJobReq) (jobId string, err error) {
 	createTime := timeutil.TimeStampNow()
@@ -136,12 +151,14 @@ func GenerateNotebookJob(ctx *context.Context, req *GenerateNotebookJobReq) (job
 			ObjectKey: req.CodeStoragePath + cloudbrain.DefaultBranchName + ".zip",
 		}
 	} else {
-
+		datasetGrampus = getDatasetGPUGrampus(req.DatasetInfos)
 		codeGrampus = models.GrampusDataset{
-			Name:      req.CodeName,
-			Bucket:    setting.Bucket,
-			EndPoint:  getEndPoint(),
-			ObjectKey: req.CodeStoragePath + cloudbrain.DefaultBranchName + ".zip",
+			Name:          req.CodeName,
+			Bucket:        setting.Attachment.Minio.Bucket,
+			EndPoint:      setting.Attachment.Minio.Endpoint,
+			ObjectKey:     req.CodeStoragePath + cloudbrain.DefaultBranchName + ".zip",
+			ReadOnly:      false,
+			ContainerPath: "/code",
 		}
 
 	}