Update R/caretPredict.R

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
zachmayer · Dec 17, 2024 · 4e7346a · 4e7346a
1 parent 8e076af
commit 4e7346a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/R/caretPredict.R b/R/caretPredict.R
@@ -86,6 +86,7 @@ caretPredict <- function(object, newdata = NULL, excluded_class_id = 1L, aggrega
 #' @return The output of the `train` function.
 #' @keywords internal
 caretTrain <- function(local_args, global_args, continue_on_fail = FALSE, trim = TRUE, aggregate_resamples = TRUE) {
+  stopifnot(is.logical(aggregate_resamples), length(aggregate_resamples) == 1L)
   # Combine args
   # I think my handling here is correct (update globals with locals, which allows locals be partial)
   # but it would be nice to have some tests