diff --git a/ext/SciMLBaseMakieExt.jl b/ext/SciMLBaseMakieExt.jl index 24c6e2c21..65acbe2c8 100644 --- a/ext/SciMLBaseMakieExt.jl +++ b/ext/SciMLBaseMakieExt.jl @@ -344,7 +344,7 @@ end # TODO: should `error_style` be Makie plot types instead? I.e. `Band`, `Errorbar`, etc function Makie.convert_arguments(::Type{<:Lines}, sim::SciMLBase.EnsembleSummary; - trajectories = sim.u.u[1] isa AbstractArray ? eachindex(sim.u.u[1]) : + idxs = sim.u.u[1] isa AbstractArray ? eachindex(sim.u.u[1]) : 1, error_style = :ribbon, ci_type = :quantile, kwargs...) @@ -384,7 +384,7 @@ function Makie.convert_arguments(::Type{<:Lines}, makie_plotlist = Makie.PlotSpec[] - for (count, idx) in enumerate(trajectories) + for (count, idx) in enumerate(idxs) push!(makie_plotlist, S.Lines(sim.t, u[idx]; color = Makie.Cycled(count), label = "u[$idx]")) if error_style == :ribbon diff --git a/src/ensemble/ensemble_solutions.jl b/src/ensemble/ensemble_solutions.jl index 7c9529b67..b2294ea25 100644 --- a/src/ensemble/ensemble_solutions.jl +++ b/src/ensemble/ensemble_solutions.jl @@ -158,7 +158,7 @@ end end @recipe function f(sim::EnsembleSummary; - trajectories = sim.u.u[1] isa AbstractArray ? eachindex(sim.u.u[1]) : + idxs = sim.u.u[1] isa AbstractArray ? eachindex(sim.u.u[1]) : 1, error_style = :ribbon, ci_type = :quantile) if ci_type == :SEM @@ -192,7 +192,7 @@ end else error("ci_type choice not valid. Must be `:SEM` or `:quantile`") end - for i in trajectories + for i in idxs @series begin legend --> false linewidth --> 3