Skip to content

Commit 0203dda

Browse files
Merge pull request #3877 from AayushSabharwal/as/fix-ode-nlstep
fix: unwrap substitution rules in `nlstep=true`
2 parents 873bf7e + 4f1233c commit 0203dda

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/systems/solver_nlprob.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,8 @@ function inner_nlsystem(sys::System, mm, nlstep_compile::Bool)
4848
outer_tmp = get_outer_tmp(N)
4949
inner_tmp = get_inner_tmp(N)
5050

51-
subrules = Dict([v => gamma2*v + inner_tmp[i] for (i, v) in enumerate(dvs)])
52-
subrules[t] = c
51+
subrules = Dict([v => unwrap(gamma2*v + inner_tmp[i]) for (i, v) in enumerate(dvs)])
52+
subrules[t] = unwrap(c)
5353
new_rhss = map(Base.Fix2(fast_substitute, subrules), rhss)
5454
new_rhss = collect(outer_tmp) .+ gamma1 .* new_rhss .- gamma3 * mm * dvs
5555
new_eqs = [0 ~ rhs for rhs in new_rhss]

0 commit comments

Comments
 (0)