add sigma entry in RegExecStats

MaxenceGollier · MaxenceGollier · commit 4bf87cb00f69 · 2025-02-23T18:44:00.000-05:00
diff --git a/src/R2_alg.jl b/src/R2_alg.jl
@@ -410,6 +410,7 @@ function SolverCore.solve!(
   set_objective!(stats, fk + hk)
   set_solver_specific!(stats, :smooth_obj, fk)
   set_solver_specific!(stats, :nonsmooth_obj, hk)
+  set_solver_specific!(stats, :sigma, σk)
 
   φk(d) = dot(∇fk, d)
   mk(d)::T = φk(d) + ψ(d)::T
@@ -498,6 +499,7 @@ function SolverCore.solve!(
     set_objective!(stats, fk + hk)
     set_solver_specific!(stats, :smooth_obj, fk)
     set_solver_specific!(stats, :nonsmooth_obj, hk)
+    set_solver_specific!(stats, :sigma, σk)
     set_iter!(stats, stats.iter + 1)
     set_time!(stats, time() - start_time)
 
diff --git a/src/utils.jl b/src/utils.jl
@@ -37,5 +37,6 @@ function RegularizedExecutionStats(reg_nlp::AbstractRegularizedNLPModel{T, V}) w
   stats = GenericExecutionStats(reg_nlp.model, solver_specific = Dict{Symbol, T}())
   set_solver_specific!(stats, :smooth_obj, T(Inf))
   set_solver_specific!(stats, :nonsmooth_obj, T(Inf))
+  set_solver_specific!(stats, :sigma, T(Inf))
   return stats
 end
