Merge pull request #3 from ParasPuneetSingh/ParasPuneetSingh-MOO-1

Updated loss function in __solve of OptimizationEvolutionary.jl
SciML · Jul 8, 2024 · 8210956 · 8210956
2 parents e987377 + 44b4a86
commit 8210956
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/lib/OptimizationEvolutionary/src/OptimizationEvolutionary.jl b/lib/OptimizationEvolutionary/src/OptimizationEvolutionary.jl
@@ -125,8 +125,13 @@ function SciMLBase.__solve(cache::OptimizationCache{
     f = cache.f
 
     _loss = function (θ)
-        x = f(θ, cache.p, cur...)
-        return first(x)
+        if isa(f, MultiObjectiveOptimizationFunction)
+            x = f(θ, cache.p, cur...)
+            return x
+        else
+            x = f(θ, cache.p, cur...)
+            return first(x)
+        end
     end
 
     opt_args = __map_optimizer_args(cache, cache.opt; callback = _cb, cache.solver_args...,