handle interp for integers by casting and rounding (invenia#71)

kpa28-git · Mar 5, 2024 · ee9856d · ee9856d
1 parent 687c56a
commit ee9856d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/imputors/interp.jl b/src/imputors/interp.jl
@@ -72,3 +72,9 @@ function _impute!(data::AbstractVector{<:Union{T, Missing}}, imp::Interpolate) w
 
     return data
 end
+
+function _impute!(data::AbstractVector{<:Union{T, Missing}}, imp::Interpolate) where {T<:Union{Signed, Unsigned}}
+    dataf = _impute!(float(data), imp)
+    data .= round.(Union{T, Missing}, dataf)
+    return data
+end