Update lib/OptimizationOptimisers/src/sophia.jl

SciML · Oct 1, 2023 · 78b398e · 78b398e
1 parent 868f4ca
commit 78b398e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/OptimizationOptimisers/src/sophia.jl b/lib/OptimizationOptimisers/src/sophia.jl
@@ -75,7 +75,7 @@ function SciMLBase.__solve(cache::OptimizationCache{
             return first(cache.f(θ, cache.p))
         elseif cache.callback == Optimization.DEFAULT_CALLBACK
             return first(cache.f(θ, cache.p, cur...))
-        elseif data == Optimization.DEFAULT_DATA
+        elseif data === Optimization.DEFAULT_DATA
             x = cache.f(θ, cache.p)
             return first(x)
         else