diff --git a/R/brm_multiple.R b/R/brm_multiple.R index 9c85c4530..149e0ff3d 100644 --- a/R/brm_multiple.R +++ b/R/brm_multiple.R @@ -268,8 +268,8 @@ is_data_list <- function(x) { is.list(x) && is.vector(x) } -warn_brmsfit_multiple <- function(x) { - if (is.brmsfit_multiple(x)) { +warn_brmsfit_multiple <- function(x, newdata = NULL) { + if (is.brmsfit_multiple(x) && is.null(newdata)) { warning2( "Using only the first imputed data set. Please interpret the results ", "with caution until a more principled approach has been implemented." diff --git a/R/extract_draws.R b/R/extract_draws.R index 674238400..14e4a9630 100644 --- a/R/extract_draws.R +++ b/R/extract_draws.R @@ -9,7 +9,7 @@ extract_draws.brmsfit <- function(x, newdata = NULL, re_formula = NULL, new_objects = list(), ...) { snl_options <- c("uncertainty", "gaussian", "old_levels") sample_new_levels <- match.arg(sample_new_levels, snl_options) - warn_brmsfit_multiple(x) + warn_brmsfit_multiple(x, newdata = newdata) x <- restructure(x) if (!incl_autocor) { x <- remove_autocor(x)