From a1370a0bf84d1f5ec1e655b2f0b37228f423773d Mon Sep 17 00:00:00 2001 From: Pepijn de Vos Date: Wed, 8 Nov 2023 12:55:22 +0100 Subject: [PATCH] put stats back This most likely got lost in a rebase --- src/ensemble/basic_ensemble_solve.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/ensemble/basic_ensemble_solve.jl b/src/ensemble/basic_ensemble_solve.jl index 97cfa88b9..ece01eb68 100644 --- a/src/ensemble/basic_ensemble_solve.jl +++ b/src/ensemble/basic_ensemble_solve.jl @@ -140,7 +140,8 @@ function __solve(prob::AbstractEnsembleProblem, elapsed_time = @elapsed u = solve_batch(prob, alg, ensemblealg, 1:trajectories, pmap_batch_size; kwargs...) _u = tighten_container_eltype(u) - return EnsembleSolution(_u, elapsed_time, true) + stats = merge_stats(_u) + return EnsembleSolution(_u, elapsed_time, true, stats) end converged::Bool = false