feat: save discretes alongside callback save

Author: AayushSabharwal

src/callbacks.jl

@@ -596,6 +596,11 @@ function apply_callback!(integrator,
 
         @inbounds if callback.save_positions[2]
             savevalues!(integrator, true)
+            if callback isa VectorContinuousCallback
+                SciMLBase.save_discretes!(integrator, callback, event_idx)
+            else
+                SciMLBase.save_discretes!(integrator, callback)
+            end
             saved_in_cb = true
         end
         return true, saved_in_cb
@@ -622,6 +627,7 @@ end
         end
         @inbounds if callback.save_positions[2]
             savevalues!(integrator, true)
+            SciMLBase.save_discretes!(integrator, callback)
             saved_in_cb = true
         end
     end