fix tests

pola-rs · May 21, 2024 · 0094c13 · 0094c13
1 parent 68348a7
commit 0094c13
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/crates/polars-io/src/csv/read/schema_inference.rs b/crates/polars-io/src/csv/read/schema_inference.rs
@@ -97,7 +97,7 @@ impl SchemaInferenceResult {
 impl CsvReadOptions {
     /// Note: This does not update the schema from the inference result.
     pub fn update_with_inference_result(&mut self, si_result: &SchemaInferenceResult) {
-        self.skip_rows += si_result.skip_rows;
+        self.skip_rows = si_result.skip_rows;
         self.n_threads = si_result.n_threads;
     }
 }

diff --git a/crates/polars-lazy/src/tests/mod.rs b/crates/polars-lazy/src/tests/mod.rs
@@ -48,7 +48,7 @@ use crate::prelude::*;
 #[cfg(feature = "parquet")]
 static GLOB_PARQUET: &str = "../../examples/datasets/*.parquet";
 #[cfg(feature = "csv")]
-static GLOB_CSV: &str = "../../examples/datasets/*.csv";
+static GLOB_CSV: &str = "../../examples/datasets/foods*.csv";
 #[cfg(feature = "ipc")]
 static GLOB_IPC: &str = "../../examples/datasets/*.ipc";
 #[cfg(feature = "parquet")]

diff --git a/crates/polars-plan/src/logical_plan/conversion/scans.rs b/crates/polars-plan/src/logical_plan/conversion/scans.rs
@@ -171,6 +171,9 @@ pub(super) fn csv_file_info(
                 let schema_a = a.get_inferred_schema();
                 let schema_b = b.get_inferred_schema();
 
+                dbg!(schema_a.clone());
+                dbg!(schema_b.clone());
+
                 match (schema_a.is_empty(), schema_b.is_empty()) {
                     (true, _) => schema_b,
                     (_, true) => schema_a,