fixes

Author: robert3005

encodings/runend/src/compute/mod.rs

@@ -5,11 +5,11 @@ pub(crate) mod filter;
 mod invert;
 mod is_sorted;
 mod min_max;
+mod nan_count;
 mod scalar_at;
 mod slice;
 pub(crate) mod take;
 mod take_from;
-mod nan_count;
 
 use vortex_array::Array;
 use vortex_array::compute::{

encodings/runend/src/compute/nan_count.rs

@@ -0,0 +1 @@
+

vortex-array/src/compute/nan_count.rs

@@ -30,20 +30,19 @@ pub fn nan_count(array: &dyn Array) -> VortexResult<Option<usize>> {
         return Ok(None);
     }
 
-    // Only float arrays can have NaNs
-    if !array.dtype().is_float() {
-        return Ok(Some(0));
-    }
-
     let nan_count = array
         .statistics()
         .get_as::<usize>(Stat::NaNCount)
         .and_then(Precision::as_exact);
+
     if let Some(nan_count) = nan_count {
         return Ok(Some(nan_count));
     }
 
-    let nan_count = if let Some(fn_) = array.vtable().nan_count_fn() {
+    // Only float arrays can have NaNs
+    let nan_count = if !array.dtype().is_float() {
+        Some(0)
+    } else if let Some(fn_) = array.vtable().nan_count_fn() {
         fn_.nan_count(array)?
     } else {
         let canonical = array.to_canonical()?;