put stats back

This most likely got lost in a rebase
SciML · Nov 8, 2023 · a1370a0 · a1370a0
1 parent 1d44a66
commit a1370a0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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