Merge pull request #218 from oscardssmith/os/preset-time-allocations-…

…redux fix remaining allocations in `PresetTimeCallback`
SciML · Apr 29, 2024 · afff8e3 · afff8e3
2 parents 1a02294 + b311715
commit afff8e3
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/src/preset_time.jl b/src/preset_time.jl
@@ -30,19 +30,23 @@ function PresetTimeCallback(tstops, user_affect!;
         else
             tstops = sort(tstops)
         end
-        condition = function (u, t, integrator)
-            if hasproperty(integrator, :dt)
-                insorted(t, tstops) && (integrator.t - integrator.dt) != integrator.t
-            else
-                insorted(t, tstops)
+        condition = let tstops = tstops
+            function (u, t, integrator)
+                if hasproperty(integrator, :dt)
+                    insorted(t, tstops) && (integrator.t - integrator.dt) != integrator.t
+                else
+                    insorted(t, tstops)
+                end
             end
         end
     elseif tstops isa Number
-        condition = function (u, t, integrator)
-            if hasproperty(integrator, :dt)
-                t == tstops && (integrator.t - integrator.dt) != integrator.t
-            else
-                t == tstops
+        condition = let tstops = tstops
+            function (u, t, integrator)
+                if hasproperty(integrator, :dt)
+                    t == tstops && (integrator.t - integrator.dt) != integrator.t
+                else
+                    t == tstops
+                end
             end
         end
     else