Update monte_solutions.jl

SciML · Jul 19, 2017 · f760365 · f760365
1 parent d2b8663
commit f760365
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/solutions/monte_solutions.jl b/src/solutions/monte_solutions.jl
@@ -116,6 +116,11 @@ end
 @recipe function f(sim::MonteCarloSummary;
                    idxs= typeof(sim.u[1])<:AbstractArray ? eachindex(sim.u[1]) : 1,
                    error_style=:ribbon,ci_type=:quantile)
+  if typeof(sim.u[1])<:AbstractArray
+    u = vecarr_to_vectors(sim.u)
+  else
+    u = [sim.u.u]
+  end
   if ci_type == :SEM
     if typeof(sim.u[1])<:AbstractArray
       ci_low = vecarr_to_vectors(VectorOfArray([sqrt(sim.v[i]/sim.num_monte).*1.96 for i in 1:length(sim.v)]))
@@ -135,11 +140,6 @@ end
   else
     error("ci_type choice not valid. Must be :variance or :quantile")
   end
-  if typeof(sim.u[1])<:AbstractArray
-    u = vecarr_to_vectors(sim.u)
-  else
-    u = [sim.u.u]
-  end
   for i in idxs
     @series begin
       legend := false