Merge pull request #599 from AayushSabharwal/as/ensemble-analysis-again

fix: fix EnsembleAnalysis methods for RAT v3 (again)
SciML · Jan 17, 2024 · 8188851 · 8188851
2 parents 4590236 + e29656f
commit 8188851
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ensemble/ensemble_analysis.jl b/src/ensemble/ensemble_analysis.jl
@@ -3,7 +3,7 @@ module EnsembleAnalysis
 using SciMLBase, Statistics, RecursiveArrayTools, StaticArraysCore
 
 # Getters
-get_timestep(sim, i) = (getindex(sol, :, i) for sol in sim)
+get_timestep(sim, i) = (sol.u[i] for sol in sim)
 get_timepoint(sim, t) = (sol(t) for sol in sim)
 function componentwise_vectors_timestep(sim, i)
     arr = [get_timestep(sim, i)...]