fix failures

SciML · Dec 27, 2023 · 7f2f357 · 7f2f357
1 parent 43abb68
commit 7f2f357
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/lib/OptimizationOptimisers/src/OptimizationOptimisers.jl b/lib/OptimizationOptimisers/src/OptimizationOptimisers.jl
@@ -74,9 +74,9 @@ function SciMLBase.__solve(cache::OptimizationCache{
             cache.progress && ProgressLogging.@logprogress msg i/maxiters
 
             if cache.solver_args.save_best
-                if first(x)[1] < first(min_err)  #found a better solution
+                if first(x)[1] < first(min_err)[1]  #found a better solution
                     min_opt = opt
-                    min_err = x[1]
+                    min_err = x
                     min_θ = copy(θ)
                 end
                 if i == maxiters  #Last iteration, revert to best.
@@ -93,7 +93,7 @@ function SciMLBase.__solve(cache::OptimizationCache{
 
     t1 = time()
 
-    SciMLBase.build_solution(cache, cache.opt, θ, x[1], solve_time = t1 - t0)
+    SciMLBase.build_solution(cache, cache.opt, θ, first(x)[1], solve_time = t1 - t0)
     # here should be build_solution to create the output message
 end
 

diff --git a/test/diffeqfluxtests.jl b/test/diffeqfluxtests.jl
@@ -87,7 +87,7 @@ function loss_neuralode(p)
 end
 
 iter = 0
-callback = function (p, l, pred)
+callback = function (p, l, pred, args...)
     global iter
     iter += 1