Merge pull request #565 from AayushSabharwal/as/pde-sii

fix: use pdesystem for AbstractPDESolution SII
SciML · Dec 21, 2023 · b699bcc · b699bcc
2 parents 97a3dff + 72a8576
commit b699bcc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/solutions/solution_interface.jl b/src/solutions/solution_interface.jl
@@ -32,6 +32,8 @@ const AbstractSolution = Union{AbstractTimeseriesSolution,AbstractNoTimeSolution
 SymbolicIndexingInterface.symbolic_container(A::AbstractSolution) = A.prob.f
 SymbolicIndexingInterface.parameter_values(A::AbstractSolution) = A.prob.p
 
+SymbolicIndexingInterface.symbolic_container(A::AbstractPDESolution) = A.disc_data.pdesys
+
 SymbolicIndexingInterface.is_independent_variable(::AbstractNoTimeSolution, sym) = false
 
 SymbolicIndexingInterface.independent_variable_symbols(::AbstractNoTimeSolution) = []