Skip to content

Commit 569dc26

Browse files
committed
fix: flatten batches in PyDataFrame to ensure proper schema conversion
1 parent 1ee861a commit 569dc26

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/dataframe.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -735,6 +735,7 @@ impl PyDataFrame {
735735

736736
batches = batches
737737
.into_iter()
738+
.flatten()
738739
.map(|record_batch| record_batch_into_schema(record_batch, &schema))
739740
.collect::<Result<Vec<RecordBatch>, ArrowError>>()
740741
.map_err(PyDataFusionError::from)?;

0 commit comments

Comments
 (0)