[R] don't throw warning when nothing changes (#11073)

dmlc · Dec 8, 2024 · 8173e57 · 8173e57
1 parent eb3a188
commit 8173e57
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R-package/R/utils.R b/R-package/R/utils.R
@@ -285,7 +285,7 @@ generate.cv.folds <- function(nfold, nrows, stratified, label, group, params) {
     return(generate.group.folds(nfold, group))
   }
   objective <- params$objective
-  if (!is.character(objective)) {
+  if (stratified && !is.character(objective)) {
     warning("Will use unstratified splitting (custom objective used)")
     stratified <- FALSE
   }