Update vfold.R

bug correction with vfold_splits function forward on breaks value from the vfold_cv function to vfold_splits
tidymodels · Apr 12, 2024 · 77350d5 · 77350d5
1 parent 030cd48
commit 77350d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/vfold.R b/R/vfold.R
@@ -86,7 +86,7 @@ vfold_cv <- function(data, v = 10, repeats = 1,
       )
     }
     for (i in 1:repeats) {
-      tmp <- vfold_splits(data = data, v = v, strata = strata, pool = pool)
+      tmp <- vfold_splits(data = data, v = v, strata = strata, breaks = breaks ,pool = pool)
       tmp$id2 <- tmp$id
       tmp$id <- names0(repeats, "Repeat")[i]
       split_objs <- if (i == 1) {