Merge pull request #888 from JuliaReach/schillic/diffeq

Fix deprecation warning in OrdinaryDiffEq
JuliaReach · Jan 9, 2025 · 112ea82 · 112ea82
2 parents 762beb4 + c5de74a
commit 112ea82
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/Initialization/init_OrdinaryDiffEq.jl b/src/Initialization/init_OrdinaryDiffEq.jl
@@ -124,9 +124,9 @@ function _solve_ensemble(ivp::InitialValueProblem{<:AbstractHybridSystem},
                 condition = (u, t, integrator) -> u ∉ I⁻ ? -1 : 1
                 callback = ODE.ContinuousCallback(condition, termination_action)
             end
-            trajectory = _solve_ensemble(ivp_loc, args...; initial_states=[x0],
-                                         callback=callback, tspan=t0 .. T,
-                                         kwargs_sim...)[1]
+            res_ens = _solve_ensemble(ivp_loc, args...; initial_states=[x0],
+                                      callback=callback, tspan=t0 .. T, kwargs_sim...)
+            trajectory = getindex(res_ens, :, 1)
 
             # remove states from trajectory if they violate invariant
             # observation: this only concerns the last two steps