Fix loo test

stan-dev · Dec 18, 2023 · 65e8cac · 65e8cac
1 parent 4fa40b6
commit 65e8cac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/fit.R b/R/fit.R
@@ -1545,7 +1545,7 @@ loo <- function(variables = "log_lik", r_eff = TRUE, moment_match = FALSE, ...)
       loo = loo_result,
       post_draws = function(x, ...) { x$draws(format = "draws_matrix") },
       log_lik_i = log_lik_i,
-      unconstrain_pars = function(x, pars, ...) { do.call(rbind, lapply(x$unconstrain_draws(), function(chain) { do.call(rbind, chain) })) },
+      unconstrain_pars = function(x, pars, ...) { x$unconstrain_draws(format = "draws_matrix") },
       log_prob_upars = function(x, upars, ...) { apply(upars, 1, x$log_prob) },
       log_lik_i_upars = log_lik_i_upars,
       ...