test: fix typo in NNODE tests

SciML · Jan 20, 2024 · 63c6733 · 63c6733
1 parent 5df70a8
commit 63c6733
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)