Skip to content

Commit b99ab7d

Browse files
fix: fix observed equations not being generated
1 parent 3e4a648 commit b99ab7d

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/systems/nonlinear/nonlinearsystem.jl

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -572,13 +572,15 @@ function SCCNonlinearFunction{iip}(
572572
_eqs = eqs[escc]
573573
obsidxs = observed_equations_used_by(sys, _eqs)
574574
_obs = obs[obsidxs]
575+
obs_assignments = [eq.lhs eq.rhs for eq in _obs]
575576

576577
cmap, cs = get_cmap(sys)
577-
assignments = [eq.lhs eq.rhs for eq in cmap]
578+
cmap_assignments = [eq.lhs eq.rhs for eq in cmap]
578579
rhss = [eq.rhs - eq.lhs for eq in _eqs]
579-
wrap_code = wrap_assignments(false, assignments) .∘
580+
wrap_code = wrap_assignments(false, cmap_assignments) .∘
580581
(wrap_array_vars(sys, rhss; dvs = _dvs, cachesyms)) .∘
581-
wrap_parameter_dependencies(sys, false)
582+
wrap_parameter_dependencies(sys, false) .∘
583+
wrap_assignments(false, obs_assignments)
582584
f_gen = build_function(
583585
rhss, _dvs, rps..., cachesyms...; wrap_code, expression = Val{true})
584586
f_oop, f_iip = eval_or_rgf.(f_gen; eval_expression, eval_module)

0 commit comments

Comments
 (0)