Update R/caretPredict.R

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
zachmayer · Dec 17, 2024 · 8e076af · 8e076af
1 parent 3e174c9
commit 8e076af
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/R/caretPredict.R b/R/caretPredict.R
@@ -13,6 +13,7 @@
 #' @keywords internal
 caretPredict <- function(object, newdata = NULL, excluded_class_id = 1L, aggregate_resamples = TRUE, ...) {
   stopifnot(methods::is(object, "train"))
+  stopifnot(is.logical(aggregate_resamples), length(aggregate_resamples) == 1L)
 
   # Extract the model type
   is_class <- isClassifierAndValidate(object, validate_for_stacking = is.null(newdata))