fixed stupid newly introduced bug (it's too late...)

boost-R · Jan 26, 2017 · 1c123af · 1c123af
1 parent 2edc5a6
commit 1c123af
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/methods.R b/R/methods.R
@@ -36,7 +36,7 @@ predict.mboost <- function(object, newdata = NULL,
     nm <- rownames(newdata)
     if (is.null(newdata)) 
         nm <- object$rownames
-    if (length(pr == 1) && length(nm) != length(pr))
+    if (length(pr) == 1 && length(nm) != length(pr))
         nm <- NULL
     if (is.list(pr)){
         RET <- lapply(pr, .predictmboost, y = object$response,
@@ -274,7 +274,7 @@ predict.glmboost <- function(object, newdata = NULL,
     nm <- rownames(newdata)
     if (is.null(newdata)) 
         nm <- object$rownames
-    if (length(pr == 1) && length(nm) != length(pr))
+    if (length(pr) == 1 && length(nm) != length(pr))
         nm <- NULL
     if (is.list(pr))
         return(lapply(pr, .predictmboost, y = object$response,