Skip to content

Commit 660edf9

Browse files
author
Sathvik Bhagavan
committed
test: update deprecated use of res.minimizer
1 parent f381923 commit 660edf9

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

test/dgm_test.jl

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -35,14 +35,14 @@ import Lux: tanh, identity
3535
end
3636

3737
res = Optimization.solve(prob, Adam(0.01); callback = callback, maxiters = 500)
38-
prob = remake(prob, u0 = res.minimizer)
38+
prob = remake(prob, u0 = res.u)
3939
res = Optimization.solve(prob, Adam(0.001); callback = callback, maxiters = 200)
4040
phi = discretization.phi
4141

4242
xs, ys = [infimum(d.domain):0.01:supremum(d.domain) for d in domains]
4343
analytic_sol_func(x, y) = (sin(pi * x) * sin(pi * y)) / (2pi^2)
4444

45-
u_predict = reshape([first(phi([x, y], res.minimizer)) for x in xs for y in ys],
45+
u_predict = reshape([first(phi([x, y], res.u)) for x in xs for y in ys],
4646
(length(xs), length(ys)))
4747
u_real = reshape([analytic_sol_func(x, y) for x in xs for y in ys],
4848
(length(xs), length(ys)))
@@ -87,7 +87,7 @@ end
8787
end
8888

8989
res = Optimization.solve(prob, Adam(0.01); callback = callback, maxiters = 300)
90-
prob = remake(prob, u0 = res.minimizer)
90+
prob = remake(prob, u0 = res.u)
9191
res = Optimization.solve(prob, Adam(0.001); callback = callback, maxiters = 300)
9292
phi = discretization.phi
9393

@@ -103,7 +103,7 @@ end
103103
xs = collect(infimum(domains2[2].domain):1.0:supremum(domains2[2].domain))
104104

105105
u_real= [analytic_sol_func(t,x) for t in ts, x in xs]
106-
u_predict= [first(phi([t, x], res.minimizer)) for t in ts, x in xs]
106+
u_predict= [first(phi([t, x], res.u)) for t in ts, x in xs]
107107
@test u_predict u_real rtol= 0.05
108108
end
109109

@@ -154,7 +154,7 @@ end
154154
res = Optimization.solve(prob, Adam(0.01); callback = callback, maxiters = 300);
155155
phi = discretization.phi;
156156

157-
u_predict= [first(phi([t, x], res.minimizer)) for t in ts, x in xs]
157+
u_predict= [first(phi([t, x], res.u)) for t in ts, x in xs]
158158

159159
@test u_predict u_MOL rtol= 0.025
160160

0 commit comments

Comments
 (0)