fix test optimisers

SciML · Jan 4, 2024 · 7e8bf0c · 7e8bf0c
1 parent d1d326a
commit 7e8bf0c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/OptimizationOptimisers/test/runtests.jl b/lib/OptimizationOptimisers/test/runtests.jl
@@ -60,8 +60,8 @@ using Zygote
         optprob = OptimizationFunction(rosenbrock, Optimization.AutoZygote())
 
         prob = OptimizationProblem(optprob, x0, _p)
-        function callback(θ, l, state, iter)
-            Optimisers.adjust!(state, 0.1/iter)
+        function callback(state, l)
+            Optimisers.adjust!(state.solver_state, 0.1/state.iter)
             return false
         end
         sol = solve(prob, Optimisers.Adam(0.1), maxiters = 1000, progress = false, callback = callback)