fix: use right integer type

dathere · Jan 8, 2025 · d8c1524 · d8c1524
1 parent 9321bf1
commit d8c1524
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/cmd/count.rs b/src/cmd/count.rs
@@ -529,10 +529,10 @@ pub fn polars_count_input(conf: &Config, low_memory: bool) -> CliResult<u64> {
         },
     };
 
-    let mut count = match sqlresult_lf.collect()?["len"].u64() {
+    let mut count = match sqlresult_lf.collect()?["len"].u32() {
         Ok(cnt) => {
             if let Some(count) = cnt.get(0) {
-                count
+                count as u64
             } else {
                 // Empty result, fall back to regular CSV reader
                 log::warn!("empty polars result, falling back to regular reader");