From d8c1524ca0dff4ac19164ccb8090b01fd740b571 Mon Sep 17 00:00:00 2001
From: Joel Natividad <1980690+jqnatividad@users.noreply.github.com>
Date: Tue, 7 Jan 2025 21:59:31 -0500
Subject: [PATCH] fix: use right integer type

---
 src/cmd/count.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/cmd/count.rs b/src/cmd/count.rs
index 80098ec85..cb3441d76 100644
--- 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");