Update TableOne.jl

markgpritchard · Mar 28, 2024 · c19a6f6 · c19a6f6
1 parent ca6d62e
commit c19a6f6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/TableOne.jl b/src/TableOne.jl
@@ -581,13 +581,13 @@ function _getvarname(var, varnames::Dict)
 end
 
 function _addnmissing!(_t, varvect, strataids)
-    n = _countmissing(varvect)
+    n = _countmissing(varvect, strataids)
     nmissing = [ "" for _ ∈ axes(_t, 1) ]
     nmissing[1] = sprint(show, n)
     insertcols!(_t, :nmissing => nmissing)
 end
 
-function _countmissing(varvect, sn=:Total)
+function _countmissing(varvect, strataids, sn=:Total)
     idmissing = findall(ismissing, varvect)
     vectorcountmissing = findall(x -> x ∈ idmissing, strataids[Symbol(sn)])
     n = length(vectorcountmissing)