fix(rust): Fix null handling in collect_statistics of parquet reader

pola-rs · Dec 19, 2024 · 16d660b · 16d660b
1 parent 2d24796
commit 16d660b
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/crates/polars-io/src/parquet/read/predicates.rs b/crates/polars-io/src/parquet/read/predicates.rs
@@ -25,12 +25,12 @@ pub(crate) fn collect_statistics(
     let stats = schema
         .iter_values()
         .map(|field| {
-            let iter = md.columns_under_root_iter(&field.name).unwrap();
-
-            Ok(if iter.len() == 0 {
-                ColumnStats::new(field.into(), None, None, None)
-            } else {
-                ColumnStats::from_arrow_stats(deserialize(field, iter)?, field)
+            let iter = md.columns_under_root_iter(&field.name);
+            Ok(match iter {
+                Some(x) if { x.len() > 0 } => {
+                    ColumnStats::from_arrow_stats(deserialize(field, x)?, field)
+                },
+                _ => ColumnStats::new(field.into(), None, None, None),
             })
         })
         .collect::<PolarsResult<Vec<_>>>()?;