add schema to recordValueWriter

ExpediaGroup · Dec 5, 2023 · 473d5eb · 473d5eb
1 parent 97c1f91
commit 473d5eb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/go/internal/feast/transformation/transformation_service.go b/go/internal/feast/transformation/transformation_service.go
@@ -63,11 +63,12 @@ func (s *GrpcTransformationService) GetTransformation(
 		inputColumns = append(inputColumns, arr)
 	}
 
-	inputRecord := array.NewRecord(arrow.NewSchema(inputFields, nil), inputColumns, int64(numRows))
+  inputSchema := arrow.NewSchema(inputFields, nil)
+	inputRecord := array.NewRecord(inputSchema, inputColumns, int64(numRows))
 	defer inputRecord.Release()
 
 	recordValueWriter := new(ByteSliceWriter)
-	arrowWriter, err := ipc.NewFileWriter(recordValueWriter)
+	arrowWriter, err := ipc.NewFileWriter(recordValueWriter, ipc.WithSchema(inputSchema))
 	if err != nil {
     fmt.Println("Error allocating NewFileWriter (GetTransformation)")
 		return nil, err