From c3214702a723f42fa6a20c61ece42a4ff1b213fb Mon Sep 17 00:00:00 2001 From: Colin Ho Date: Thu, 10 Oct 2024 16:14:45 -0700 Subject: [PATCH] fix --- src/daft-parquet/src/stream_reader.rs | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/daft-parquet/src/stream_reader.rs b/src/daft-parquet/src/stream_reader.rs index fd77efe88..58a3662cf 100644 --- a/src/daft-parquet/src/stream_reader.rs +++ b/src/daft-parquet/src/stream_reader.rs @@ -111,8 +111,12 @@ pub fn arrow_column_iters_to_table_iter( } index_so_far += len; - let mut table = Table::new_with_size(owned_schema_ref.clone(), all_series, len) - .with_context(|_| super::UnableToCreateTableFromChunkSnafu { path: uri.clone() })?; + let mut table = Table::new_with_size( + Schema::new(all_series.iter().map(|s| s.field().clone()).collect())?, + all_series, + len, + ) + .with_context(|_| super::UnableToCreateTableFromChunkSnafu { path: uri.clone() })?; // Apply pushdowns if needed if let Some(predicate) = &predicate { table = table.filter(&[predicate.clone()])?;