Skip to content

Commit 028ab9d

Browse files
committed
Remove unnecessary closures
1 parent fc3bad5 commit 028ab9d

File tree

1 file changed

+5
-10
lines changed

1 file changed

+5
-10
lines changed

src/ensemble/basic_ensemble_solve.jl

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,9 @@ function __solve(prob::AbstractEnsembleProblem,
7575
__solve(prob,alg,ensemblealg;trajectories=trajectories,kwargs...)
7676
end
7777

78+
tighten_container_eltype(u::Vector{Any}) = map(identity, u)
79+
tighten_container_eltype(u) = u
80+
7881
function __solve(prob::AbstractEnsembleProblem,
7982
alg::Union{DEAlgorithm,Nothing},
8083
ensemblealg::BasicEnsembleAlgorithm;
@@ -90,11 +93,7 @@ function __solve(prob::AbstractEnsembleProblem,
9093

9194
if num_batches == 1 && prob.reduction === DEFAULT_REDUCTION
9295
elapsed_time = @elapsed u = batch_function(1:trajectories)
93-
if typeof(u) <: Vector{Any}
94-
_u = map(i->u[i],1:length(u))
95-
else
96-
_u = u
97-
end
96+
_u = tighten_container_eltype(u)
9897
return EnsembleSolution(_u,elapsed_time,true)
9998
end
10099

@@ -118,11 +117,7 @@ function __solve(prob::AbstractEnsembleProblem,
118117
converged && break
119118
end
120119

121-
if typeof(u) <: Vector{Any}
122-
_u = map(i->u[i],1:length(u))
123-
else
124-
_u = u
125-
end
120+
_u = tighten_container_eltype(u)
126121

127122
return EnsembleSolution(u,elapsed_time,converged)
128123
end

0 commit comments

Comments
 (0)