make a bit safer

Author: ChrisRackauckas

src/ensemble/basic_ensemble_solve.jl

@@ -89,8 +89,13 @@ function __solve(prob::AbstractEnsembleProblem,
   end
 
   if num_batches == 1 && prob.reduction === DEFAULT_REDUCTION
-    elapsed_time = @elapsed batch_data = batch_function(1:trajectories)
-    return EnsembleSolution(batch_data,elapsed_time,true)
+    elapsed_time = @elapsed u = batch_function(1:trajectories)
+    if typeof(u) <: Vector{Any}
+      _u = map(i->u[i],1:length(u))
+    else
+      _u = u
+    end
+    return EnsembleSolution(_u,elapsed_time,true)
   end
 
   if prob.u_init === nothing && prob.reduction === DEFAULT_REDUCTION

src/ensemble/ensemble_problems.jl

@@ -12,7 +12,7 @@ end
 
 DEFAULT_PROB_FUNC(prob,i,repeat) = prob
 DEFAULT_OUTPUT_FUNC(sol,i) = (sol,false)
-DEFAULT_REDUCTION(u,data,I) = (reduce(vcat,(u,data)),false)
+DEFAULT_REDUCTION(u,data,I) = (vcat(u,data),false)
 EnsembleProblem(prob;
     output_func = DEFAULT_OUTPUT_FUNC,
     prob_func= DEFAULT_PROB_FUNC,