diff --git a/src/problems/problem_interface.jl b/src/problems/problem_interface.jl index a4213c906..7fe35848f 100644 --- 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))