@@ -75,6 +75,9 @@ function __solve(prob::AbstractEnsembleProblem,
75
75
__solve (prob,alg,ensemblealg;trajectories= trajectories,kwargs... )
76
76
end
77
77
78
+ tighten_container_eltype (u:: Vector{Any} ) = map (identity, u)
79
+ tighten_container_eltype (u) = u
80
+
78
81
function __solve (prob:: AbstractEnsembleProblem ,
79
82
alg:: Union{DEAlgorithm,Nothing} ,
80
83
ensemblealg:: BasicEnsembleAlgorithm ;
@@ -90,11 +93,7 @@ function __solve(prob::AbstractEnsembleProblem,
90
93
91
94
if num_batches == 1 && prob. reduction === DEFAULT_REDUCTION
92
95
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)
98
97
return EnsembleSolution (_u,elapsed_time,true )
99
98
end
100
99
@@ -118,11 +117,7 @@ function __solve(prob::AbstractEnsembleProblem,
118
117
converged && break
119
118
end
120
119
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)
126
121
127
122
return EnsembleSolution (u,elapsed_time,converged)
128
123
end
0 commit comments