correctly check all predictor variables are in new data

tiemvanderdeure · Mar 5, 2024 · c59bb97 · c59bb97
1 parent 574b696
commit c59bb97
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/predict.jl b/src/predict.jl
@@ -27,8 +27,8 @@ prediction = Maxnet.predict(bradypus_model, env)
 """
 function predict(m::MaxnetModel, x; link = CloglogLink(), clamp = false)
     predictors = Tables.columntable(x)
-    for k in keys(predictors)
-        k in keys(m.predictor_data) || error("$k is not found in the predictors")
+    for k in keys(m.predictor_data)
+        k in keys(predictors) || error("$k is not found in the predictors")
     end
 
     # clamp the predictors