|
|
|
@@ -1178,22 +1178,18 @@ func getDatasetDownloadInfo(ctx *context.Context, task *models.Cloudbrain) []mod |
|
|
|
if task.IsGPUTask() { |
|
|
|
return GetCloudBrainDataSetInfo(task.Uuid, task.DatasetName, false) |
|
|
|
} else { |
|
|
|
datasetDownload = GetCloudBrainDataSetInfo(task.Uuid, task.DatasetName, true) |
|
|
|
datasetDownload = GetCloudBrainDataSetInfo(task.Uuid, task.DatasetName, false) |
|
|
|
datasetObsUrlList := make([]NotebookDataset, 0) |
|
|
|
_ = json.Unmarshal([]byte(task.DataUrl), &datasetObsUrlList) |
|
|
|
|
|
|
|
for _, datasetInfo := range datasetDownload { |
|
|
|
match := false |
|
|
|
|
|
|
|
for _, datasetObs := range datasetObsUrlList { |
|
|
|
if strings.Contains(datasetObs.DatasetUrl, datasetInfo.DatasetName) { |
|
|
|
datasetInfo.DatasetDownloadLink = datasetObs.DatasetUrl |
|
|
|
match = true |
|
|
|
break |
|
|
|
} |
|
|
|
} |
|
|
|
if !match { |
|
|
|
datasetInfo.DatasetDownloadLink = "" |
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|