fix limit on single threaded parquet reader

pola-rs · Dec 22, 2021 · 1d848c4 · 1d848c4
1 parent fab93e6
commit 1d848c4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/polars/polars-arrow/src/io/parquet.rs b/polars/polars-arrow/src/io/parquet.rs
@@ -48,14 +48,15 @@ pub fn read_parquet<R: Read + Seek>(
                 if array.len() > remaining_rows {
                     array = array.slice(0, remaining_rows);
                 }
-                remaining_rows -= array.len();
 
                 buf_1 = b1;
                 buf_2 = b2;
 
                 Ok(Arc::from(array))
             })
             .collect::<Result<Vec<_>>>()?;
+
+        remaining_rows = metadata.row_groups[rg].num_rows() as usize;
         rb.push(arrs)
     }