diff --git a/R/fit.R b/R/fit.R index 498dbd295..0bffa1cc5 100644 --- a/R/fit.R +++ b/R/fit.R @@ -560,10 +560,10 @@ unconstrain_draws <- function(files = NULL, draws = NULL) { generated_quantities = FALSE) par_columns <- !(names(draws) %in% c(".chain", ".iteration", ".draw")) unconstrained <- lapply(split(draws, f = draws$.chain), function(chain) { - apply(chain, 1, function(draw) { + lapply(asplit(chain, 1), function(draw) { par_list <- utils::relist(as.numeric(draw[par_columns]), skeleton) self$unconstrain_variables(variables = par_list) - }, simplify = FALSE) + }) }) } CmdStanFit$set("public", name = "unconstrain_draws", value = unconstrain_draws)