Skip to content

Commit ab32baa

Browse files
fix: fix observed equations not being generated
1 parent 27113f2 commit ab32baa

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
@@ -568,13 +568,15 @@ function SCCNonlinearFunction{iip}(
568568
_eqs = eqs[escc]
569569
obsidxs = observed_equations_used_by(sys, _eqs)
570570
_obs = obs[obsidxs]
571+
obs_assignments = [eq.lhs eq.rhs for eq in _obs]
571572

572573
cmap, cs = get_cmap(sys)
573-
assignments = [eq.lhs eq.rhs for eq in cmap]
574+
cmap_assignments = [eq.lhs eq.rhs for eq in cmap]
574575
rhss = [eq.rhs - eq.lhs for eq in _eqs]
575-
wrap_code = wrap_assignments(false, assignments) .∘
576+
wrap_code = wrap_assignments(false, cmap_assignments) .∘
576577
(wrap_array_vars(sys, rhss; dvs = _dvs, cachesyms)) .∘
577-
wrap_parameter_dependencies(sys, false)
578+
wrap_parameter_dependencies(sys, false) .∘
579+
wrap_assignments(false, obs_assignments)
578580
f_gen = build_function(
579581
rhss, _dvs, rps..., cachesyms...; wrap_code, expression = Val{true})
580582
f_oop, f_iip = eval_or_rgf.(f_gen; eval_expression, eval_module)

0 commit comments

Comments
 (0)