Update R-package/R/xgboost.R

Co-authored-by: Michael Mayer <[email protected]>
dmlc · Jun 29, 2024 · a4df40f · a4df40f
1 parent 057bd23
commit a4df40f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R-package/R/xgboost.R b/R-package/R/xgboost.R
@@ -69,7 +69,7 @@ process.base.margin <- function(base_margin, nrows, ncols) {
 
   if (ncols == 1L) {
     if (inherits(base_margin, c("matrix", "data.frame"))) {
-      if (ncol(base_margin) != 1) {
+      if (ncol(base_margin) != 1L) {
         stop("'base_margin' should be a 1-d vector for the given objective and data.")
       }
       if (is.data.frame(base_margin)) {