Skip to content

Commit 83295d1

Browse files
improve mk syntax in R2N
1 parent 02f120d commit 83295d1

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

src/R2N.jl

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -340,13 +340,13 @@ function SolverCore.solve!(
340340
d -> φ1(d) + ψ(d)::T
341341
end
342342

343-
mk = let ψ = ψ, solver = solver
343+
mk = let ψ = ψ, model = solver.subpb.model
344344
d -> begin
345-
temp_σ = solver.subpb.model.data.σ
346-
solver.subpb.model.data.σ = zero(T)
347-
smooth_obj = obj(solver.subpb.model, d)
348-
solver.subpb.model.data.σ = temp_σ
349-
return smooth_obj + ψ(d)
345+
temp_σ = model.data.σ
346+
model.data.σ = zero(T)
347+
obj = obj(model, d) + ψ(d)
348+
model.data.σ = temp_σ
349+
return obj
350350
end
351351
end
352352

0 commit comments

Comments
 (0)