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,