From 597e19c066fc915ca9ed57d2eae4f02f530cdb0c Mon Sep 17 00:00:00 2001 From: chenyifan01 Date: Thu, 20 Jul 2023 18:01:10 +0800 Subject: [PATCH] fix bug --- services/ai_task_service/cluster/c2net.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/services/ai_task_service/cluster/c2net.go b/services/ai_task_service/cluster/c2net.go index a11dd34a1d..6e49ff1720 100644 --- a/services/ai_task_service/cluster/c2net.go +++ b/services/ai_task_service/cluster/c2net.go @@ -172,12 +172,16 @@ func getCopyCmd(datasetName, repoName, bootfilepath string) string { } func convertNoteBookTask2Grampus(t entity.NoteBookTask, command string) models.GrampusNotebookTask { - code := models.GrampusDataset{} codeArray := convertContainerArray2GrampusArray(t.Code) if codeArray != nil && len(codeArray) > 0 { code = codeArray[0] } + output := models.GrampusDataset{} + outputArray := convertContainerArray2GrampusArray(t.OutPut) + if outputArray != nil && len(outputArray) > 0 { + output = outputArray[0] + } return models.GrampusNotebookTask{ Name: t.Name, ResourceSpecId: t.Spec.SourceSpecId, @@ -185,6 +189,7 @@ func convertNoteBookTask2Grampus(t entity.NoteBookTask, command string) models.G ImageUrl: t.ImageUrl, Datasets: convertContainerArray2GrampusArray(t.Datasets), Code: code, + OutPut: output, AutoStopDuration: t.AutoStopDuration, Capacity: t.Capacity, Command: command, -- 2.34.1