fix: Merge Parquet nested and flat decoders (#21342)

pola-rs · Feb 19, 2025 · d6babcf · d6babcf
1 parent d6bb315
commit d6babcf
Show file tree

Hide file tree

Showing 6 changed files with 264 additions and 447 deletions.
diff --git a/crates/polars-parquet/src/arrow/read/deserialize/mod.rs b/crates/polars-parquet/src/arrow/read/deserialize/mod.rs
@@ -136,8 +136,13 @@ fn columns_to_iter_recursive(
     filter: Option<Filter>,
 ) -> ParquetResult<(NestedState, Box<dyn Array>, Bitmap)> {
     if init.is_empty() && is_primitive(&field.dtype) {
-        let (array, pred_true_mask) =
-            page_iter_to_array(columns.pop().unwrap(), types.pop().unwrap(), field, filter)?;
+        let (_, array, pred_true_mask) = page_iter_to_array(
+            columns.pop().unwrap(),
+            types.pop().unwrap(),
+            field,
+            filter,
+            None,
+        )?;
 
         return Ok((NestedState::default(), array, pred_true_mask));
     }