fix row count

pola-rs · Oct 22, 2023 · 975d6fb · 975d6fb
1 parent 52d63d8
commit 975d6fb
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/crates/polars-lazy/src/physical_plan/executors/scan/parquet.rs b/crates/polars-lazy/src/physical_plan/executors/scan/parquet.rs
@@ -80,7 +80,7 @@ impl ParquetExec {
                         &self.predicate,
                         &mut self.file_options.with_columns.clone(),
                         &mut self.file_info.schema.clone(),
-                        self.file_options.row_count.is_some(),
+                        base_row_count.is_some(),
                         hive_partitions.as_deref(),
                     );
 
@@ -138,9 +138,8 @@ impl ParquetExec {
                     .collect::<PolarsResult<Vec<_>>>()
             })?;
 
-            let n_read =
-                remaining_rows_to_read.saturating_sub(out.iter().map(|df| df.height()).sum());
-            remaining_rows_to_read = n_read;
+            let n_read = out.iter().map(|df| df.height()).sum();
+            remaining_rows_to_read = remaining_rows_to_read.saturating_sub(n_read);
             if let Some(rc) = &mut base_row_count {
                 rc.offset += n_read as IdxSize;
             }