From 799f7de3fa1d426037ad05618a4d5bcc0b5709df Mon Sep 17 00:00:00 2001 From: Vaibhav Dixit Date: Fri, 20 Sep 2024 16:48:36 -0400 Subject: [PATCH] Fix random tests --- docs/pages.jl | 4 ++-- docs/src/tutorials/minibatch.md | 8 +------- lib/OptimizationNLopt/test/runtests.jl | 3 +-- test/diffeqfluxtests.jl | 1 + 4 files changed, 5 insertions(+), 11 deletions(-) diff --git a/docs/pages.jl b/docs/pages.jl index 83d03f539..baeca2800 100644 --- a/docs/pages.jl +++ b/docs/pages.jl @@ -7,7 +7,7 @@ pages = ["index.md", "tutorials/linearandinteger.md", "tutorials/minibatch.md", "tutorials/remakecomposition.md", - "tutorials/symbolic.md", + "tutorials/symbolic.md" ], "Examples" => [ "examples/rosenbrock.md" @@ -41,6 +41,6 @@ pages = ["index.md", "Polyalgorithms.jl" => "optimization_packages/polyopt.md", "PRIMA.jl" => "optimization_packages/prima.md", "QuadDIRECT.jl" => "optimization_packages/quaddirect.md", - "SpeedMapping.jl" => "optimization_packages/speedmapping.md", + "SpeedMapping.jl" => "optimization_packages/speedmapping.md" ] ] diff --git a/docs/src/tutorials/minibatch.md b/docs/src/tutorials/minibatch.md index 7572b7819..08f362f71 100644 --- a/docs/src/tutorials/minibatch.md +++ b/docs/src/tutorials/minibatch.md @@ -32,14 +32,8 @@ function dudt_(u, p, t) smodel(u, p) .* u end -function callback(state, l, pred; doplot = false) #callback function to observe training +function callback(state, l) #callback function to observe training display(l) - # plot current prediction against data - if doplot - pl = scatter(t, ode_data[1, :], label = "data") - scatter!(pl, t, pred[1, :], label = "prediction") - display(plot(pl)) - end return false end diff --git a/lib/OptimizationNLopt/test/runtests.jl b/lib/OptimizationNLopt/test/runtests.jl index 5245dfe50..65c91372d 100644 --- a/lib/OptimizationNLopt/test/runtests.jl +++ b/lib/OptimizationNLopt/test/runtests.jl @@ -117,8 +117,7 @@ using Test, Random # @test sol.retcode == ReturnCode.Success @test 10 * sol.objective < l1 - Random.seed!(1) - prob = OptimizationProblem(optprob, rand(2), _p, lcons = [-Inf, -Inf], + prob = OptimizationProblem(optprob, [0.5, 0.5], _p, lcons = [-Inf, -Inf], ucons = [0.0, 0.0], lb = [-1.0, -1.0], ub = [1.0, 1.0]) sol = solve(prob, NLopt.GN_ISRES(), maxiters = 1000) @test sol.retcode == ReturnCode.MaxIters diff --git a/test/diffeqfluxtests.jl b/test/diffeqfluxtests.jl index c92463ba0..243027246 100644 --- a/test/diffeqfluxtests.jl +++ b/test/diffeqfluxtests.jl @@ -1,6 +1,7 @@ using OrdinaryDiffEq, DiffEqFlux, Lux, Optimization, OptimizationOptimJL, OptimizationOptimisers, ForwardDiff, ComponentArrays, Random rng = Random.default_rng() +Random.seed!(123) function lotka_volterra!(du, u, p, t) x, y = u