Merge ad4d693 into 888a36b

SciML · Apr 4, 2023 · af92fbb · af92fbb
2 parents 888a36b + ad4d693
commit af92fbb
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/JumpProcesses.jl b/src/JumpProcesses.jl
@@ -7,7 +7,7 @@ using RandomNumbers, TreeViews, LinearAlgebra, Markdown, DocStringExtensions
 using DataStructures, PoissonRandom, Random, ArrayInterface
 using FunctionWrappers, UnPack
 using Graphs
-using SciMLBase: SciMLBase
+using SciMLBase: SciMLBase, ___internal_setindex!
 using Base.FastMath: add_fast
 
 import DiffEqBase: DiscreteCallback, init, solve, solve!, plot_indices, initialize!

diff --git a/src/problem.jl b/src/problem.jl
@@ -119,6 +119,19 @@ function DiffEqBase.remake(thing::JumpProblem; kwargs...)
       thing.kwargs)
 end
 
+# when setindex! is used.
+function Base.setindex!(prob::JumpProblem, args...; kwargs...)
+    ___internal_setindex!(prob.prob, args...; kwargs...)
+    if using_params(prob.massaction_jump)
+        update_parameters!(prob.massaction_jump, prob.prob.p)
+    end
+end
+
+# when getindex is used.
+function Base.getindex(prob::JumpProblem, args...; kwargs...)
+    Base.getindex(prob.prob, args...; kwargs...)
+end
+
 DiffEqBase.isinplace(::JumpProblem{iip}) where {iip} = iip
 JumpProblem(prob::JumpProblem) = prob