find_predictable bug fix

yufongpeng · Dec 21, 2023 · 6c862de · 6c862de
1 parent 79410b9
commit 6c862de
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/rt.jl b/src/rt.jl
@@ -343,6 +343,9 @@ function find_predictable(model, tbl)
  level = sch[Term(name)].contrasts.levels
  union!(todel, findall(!in(level), getproperty(tbl, name)))
  end
+ for name in setdiff(propertynames(tbl), toval)
+ union!(todel, findall(isnothing, getproperty(tbl, name)))
+ end
  setdiff(eachindex(tbl), todel)
 end
 """