Skip to content

Commit 4f1233c

Browse files
fix: unwrap substitution rules in nlstep=true
1 parent 873bf7e commit 4f1233c

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)