fixes decompress_impl for csv with n_rows set (#17118)

pola-rs · Jun 22, 2024 · 3af8115 · 3af8115
1 parent 7f5412a
commit 3af8115
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/crates/polars-io/src/csv/read/utils.rs b/crates/polars-io/src/csv/read/utils.rs
@@ -129,6 +129,9 @@ fn decompress_impl<R: Read>(
                     },
                 };
             }
+            if line_count == n_rows {
+                out.truncate(buf_pos); // retain only first n_rows in out
+            }
             out
         },
     })