@@ -2528,16 +2528,18 @@ function extend(sys::AbstractSystem, basesys::AbstractSystem; name::Symbol = nam
2528
2528
cevs = union (get_continuous_events (basesys), get_continuous_events (sys))
2529
2529
devs = union (get_discrete_events (basesys), get_discrete_events (sys))
2530
2530
defs = merge (get_defaults (basesys), get_defaults (sys)) # prefer `sys`
2531
+ ieqs = union (get_initialization_eqs (basesys), get_initialization_eqs (sys))
2531
2532
syss = union (get_systems (basesys), get_systems (sys))
2532
2533
2533
2534
if length (ivs) == 0
2534
2535
T (eqs, sts, ps, observed = obs, defaults = defs, name = name, systems = syss,
2535
2536
continuous_events = cevs, discrete_events = devs, gui_metadata = gui_metadata,
2536
- parameter_dependencies = dep_ps)
2537
+ parameter_dependencies = dep_ps, initialization_eqs = ieqs )
2537
2538
elseif length (ivs) == 1
2538
2539
T (eqs, ivs[1 ], sts, ps, observed = obs, defaults = defs, name = name,
2539
2540
systems = syss, continuous_events = cevs, discrete_events = devs,
2540
- gui_metadata = gui_metadata, parameter_dependencies = dep_ps)
2541
+ gui_metadata = gui_metadata, parameter_dependencies = dep_ps,
2542
+ initialization_eqs = ieqs)
2541
2543
end
2542
2544
end
2543
2545
0 commit comments