fix: add events to SDESystem after structural simplification

Author: vyudu

src/systems/diffeqs/odesystem.jl

@@ -312,7 +312,7 @@ function ODESystem(deqs::AbstractVector{<:Equation}, iv, dvs, ps;
     end
 
     algeeqs = filter(eq -> eq.lhs isa Union{Symbolic, Number} && !is_diff_equation(eq),
-        flatten_equations(deqs))
+        deqs)
     cont_callbacks = SymbolicContinuousCallbacks(continuous_events; algeeqs, iv)
     disc_callbacks = SymbolicDiscreteCallbacks(discrete_events; algeeqs, iv)
 

src/systems/diffeqs/sdesystem.jl

@@ -269,7 +269,7 @@ function SDESystem(deqs::AbstractVector{<:Equation}, neqs::AbstractArray, iv, dv
     Wfact_t = RefValue(EMPTY_JAC)
 
     algeeqs = filter(eq -> eq.lhs isa Union{Symbolic, Number} && !is_diff_equation(eq),
-        flatten_equations(deqs))
+        deqs)
     cont_callbacks = SymbolicContinuousCallbacks(continuous_events; algeeqs, iv)
     disc_callbacks = SymbolicDiscreteCallbacks(discrete_events; algeeqs, iv)
     if is_dde === nothing

src/systems/systems.jl

@@ -155,7 +155,7 @@ function __structural_simplify(sys::AbstractSystem, io = nothing; simplify = fal
             get_iv(ode_sys), unknowns(ode_sys), parameters(ode_sys);
             name = nameof(ode_sys), is_scalar_noise, observed = observed(ode_sys), defaults = defaults(sys),
             parameter_dependencies = parameter_dependencies(sys), assertions = assertions(sys),
-            guesses = guesses(sys), initialization_eqs = initialization_equations(sys))
+            guesses = guesses(sys), initialization_eqs = initialization_equations(sys), continuous_events = continuous_events(sys), discrete_events = discrete_events(sys))
     end
 end