Update xgboostImpute.R - fix bug in input check when using objective

statistikat · Jun 13, 2024 · c5ada60 · c5ada60
1 parent 627672a
commit c5ada60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/xgboostImpute.R b/R/xgboostImpute.R
@@ -40,7 +40,7 @@ xgboostImpute <- function(formula, data, imp_var = TRUE,
   rhs_na <- apply(subset(data, select = rhs2), 1, function(x) any(is.na(x)))
   #objective should be a vector of lenght equal to the lhs variables
   if(!is.null(objective)){
-    stopifnot(length(objective)!=length(lhs))
+    stopifnot(length(objective)==length(lhs))
   }
   for (lhsV in lhs) {
     form <- as.formula(paste(lhsV, "~", rhs,"-1"))