refactor: fix solution indexing when it isn't a VectorOfArray

SciML · Nov 29, 2023 · 7a9aadb · 7a9aadb
1 parent d1db9b7
commit 7a9aadb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/solutions/solution_interface.jl b/src/solutions/solution_interface.jl
@@ -58,6 +58,10 @@ SymbolicIndexingInterface.is_time_dependent(::AbstractNoTimeSolution) = false
 # TODO make this nontrivial once dynamic state selection works
 SymbolicIndexingInterface.constant_structure(::AbstractSolution) = true
 
+Base.@propagate_inbounds function Base.getindex(A::AbstractTimeseriesSolution, ::Colon)
+    return A.u[:]
+end
+
 Base.@propagate_inbounds function Base.getindex(A::AbstractNoTimeSolution, sym)
     if symbolic_type(sym) == ScalarSymbolic()
         if is_variable(A, sym)