Skip to content

Commit e7eddb7

Browse files
Vaibhavdixit02ChrisRackauckas
authored andcommitted
format
1 parent 9d70acd commit e7eddb7

File tree

2 files changed

+11
-10
lines changed

2 files changed

+11
-10
lines changed

src/systems/optimization/optimizationsystem.jl

+6-6
Original file line numberDiff line numberDiff line change
@@ -365,11 +365,10 @@ function DiffEqBase.OptimizationProblem{iip}(sys::OptimizationSystem, u0map,
365365
expression = Val{false})
366366
if cons_j
367367
_cons_j = let (cons_jac_oop, cons_jac_iip) = generate_jacobian(cons_sys;
368-
checkbounds = checkbounds,
369-
linenumbers = linenumbers,
370-
parallel = parallel, expression = Val{false},
371-
sparse = cons_sparse)
372-
368+
checkbounds = checkbounds,
369+
linenumbers = linenumbers,
370+
parallel = parallel, expression = Val{false},
371+
sparse = cons_sparse)
373372
_cons_j(u, p) = cons_jac_oop(u, p)
374373
_cons_j(J, u, p) = (cons_jac_iip(J, u, p); J)
375374
_cons_j(u, p::MTKParameters) = cons_jac_oop(u, p...)
@@ -380,7 +379,8 @@ function DiffEqBase.OptimizationProblem{iip}(sys::OptimizationSystem, u0map,
380379
_cons_j = nothing
381380
end
382381
if cons_h
383-
_cons_h = let (cons_hess_oop, cons_hess_iip) = generate_hessian(cons_sys, checkbounds = checkbounds,
382+
_cons_h = let (cons_hess_oop, cons_hess_iip) = generate_hessian(
383+
cons_sys, checkbounds = checkbounds,
384384
linenumbers = linenumbers,
385385
sparse = cons_sparse, parallel = parallel,
386386
expression = Val{false})

test/optimizationsystem.jl

+5-4
Original file line numberDiff line numberDiff line change
@@ -318,18 +318,19 @@ end
318318
@parameters a b
319319
loss = (a - x)^2 + b * (y - x^2)^2
320320
cons2 = [x^2 + y^2 ~ 0, y * sin(x) - x ~ 0]
321-
sys = complete(OptimizationSystem(loss, [x, y], [a, b], name = :sys2, constraints = cons2))
321+
sys = complete(OptimizationSystem(
322+
loss, [x, y], [a, b], name = :sys2, constraints = cons2))
322323
prob = OptimizationProblem(sys, [x => 0.0, y => 0.0], [a => 1.0, b => 100.0],
323324
grad = true, hess = true, cons_j = true, cons_h = true)
324325

325326
G1 = Array{Float64}(undef, 2)
326327
H1 = Array{Float64}(undef, 2, 2)
327328
J = Array{Float64}(undef, 2, 2)
328329
H3 = [Array{Float64}(undef, 2, 2), Array{Float64}(undef, 2, 2)]
329-
330+
330331
prob.f.grad(G1, [1.0, 1.0], [1.0, 100.0])
331332
@test prob.f.grad([1.0, 1.0], [1.0, 100.0]) == G1
332-
333+
333334
prob.f.hess(H1, [1.0, 1.0], [1.0, 100.0])
334335
@test prob.f.hess([1.0, 1.0], [1.0, 100.0]) == H1
335336

@@ -338,4 +339,4 @@ end
338339

339340
prob.f.cons_h(H3, [1.0, 1.0], [1.0, 100.0])
340341
@test prob.f.cons_h([1.0, 1.0], [1.0, 100.0]) == H3
341-
end
342+
end

0 commit comments

Comments
 (0)