diff --git a/crates/polars-stream/src/nodes/parquet_source/row_group_decode.rs b/crates/polars-stream/src/nodes/parquet_source/row_group_decode.rs index eda18101d1a0..057d9e15d1f6 100644 --- a/crates/polars-stream/src/nodes/parquet_source/row_group_decode.rs +++ b/crates/polars-stream/src/nodes/parquet_source/row_group_decode.rs @@ -645,11 +645,15 @@ fn decode_column_prefiltered( let column = Series::try_from((arrow_field, array))?.into_column(); - if !prefilter { - column.filter(mask) + let column = if !prefilter { + column.filter(mask)? } else { - Ok(column) - } + column + }; + + assert_eq!(column.len(), expected_num_rows); + + Ok(column) } mod tests {