fix: properly implement SII fallbacks for AbstractJumpProblem

SciML · Feb 19, 2024 · 8fda165 · 8fda165
1 parent 69c7776
commit 8fda165
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/problems/problem_interface.jl b/src/problems/problem_interface.jl
@@ -6,10 +6,14 @@ Base.@propagate_inbounds function Base.getproperty(prob::AbstractSciMLProblem, s
 end
 
 SymbolicIndexingInterface.symbolic_container(prob::AbstractSciMLProblem) = prob.f
+SymbolicIndexingInterface.symbolic_container(prob::AbstractJumpProblem) = prob.prob
 
 SymbolicIndexingInterface.parameter_values(prob::AbstractSciMLProblem) = prob.p
+SymbolicIndexingInterface.parameter_values(prob::AbstractJumpProblem) = prob.prob.p
 SymbolicIndexingInterface.state_values(prob::AbstractSciMLProblem) = prob.u0
+SymbolicIndexingInterface.state_values(prob::AbstractJumpProblem) = prob.prob.u0
 SymbolicIndexingInterface.current_time(prob::AbstractSciMLProblem) = prob.tspan[1]
+SymbolicIndexingInterface.current_time(prob::AbstractJumpProblem) = prob.prob.tspan[1]
 
 Base.@propagate_inbounds function Base.getindex(prob::AbstractSciMLProblem, ::SymbolicIndexingInterface.SolvedVariables)
     return getindex(prob, variable_symbols(prob))