From 63c6733abe112abfd60498de3e838e37c34f837e Mon Sep 17 00:00:00 2001 From: Sathvik Bhagavan Date: Sat, 20 Jan 2024 15:35:28 +0000 Subject: [PATCH] test: fix typo in NNODE tests --- test/NNODE_tests.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/NNODE_tests.jl b/test/NNODE_tests.jl index 0ea1985267..e291c95306 100644 --- a/test/NNODE_tests.jl +++ b/test/NNODE_tests.jl @@ -17,7 +17,7 @@ opt = OptimizationOptimisers.Adam(0.1, (0.9, 0.95)) sol = solve(prob, NeuralPDE.NNODE(chain, opt), dt = 1 / 20.0f0, verbose = true, abstol = 1.0f-10, maxiters = 200) -@test_throws Any solve(prob, NeuralPDE.NNODE(chain, opt; autodiff = true), dt = 1 / 20.0f0, +solve(prob, NeuralPDE.NNODE(chain, opt; autodiff = true), dt = 1 / 20.0f0, verbose = true, abstol = 1.0f-10, maxiters = 200) sol = solve(prob, NeuralPDE.NNODE(chain, opt), verbose = true, @@ -26,7 +26,7 @@ sol = solve(prob, NeuralPDE.NNODE(chain, opt), verbose = true, sol = solve(prob, NeuralPDE.NNODE(luxchain, opt), dt = 1 / 20.0f0, verbose = true, abstol = 1.0f-10, maxiters = 200) -Any solve(prob, NeuralPDE.NNODE(luxchain, opt; autodiff = true), +solve(prob, NeuralPDE.NNODE(luxchain, opt; autodiff = true), dt = 1 / 20.0f0, verbose = true, abstol = 1.0f-10, maxiters = 200)