From 42017ae346dac159e345a141a7357362a30efdf1 Mon Sep 17 00:00:00 2001 From: Oscar Dowson Date: Wed, 6 Nov 2024 10:22:33 +1300 Subject: [PATCH] Fix flakey tests (#161) --- test/test_Flux.jl | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/test/test_Flux.jl b/test/test_Flux.jl index 79c4ffce..74f2a673 100644 --- a/test/test_Flux.jl +++ b/test/test_Flux.jl @@ -190,7 +190,7 @@ function test_gray_box_scalar_output() chain = Flux.Chain(Flux.Dense(2 => 16, Flux.relu), Flux.Dense(16 => 1)) model = Model(Ipopt.Optimizer) set_silent(model) - set_attribute(model, "max_iter", 5) + set_attribute(model, "max_iter", 1) @variable(model, 0 <= x[1:2] <= 1) y, formulation = MathOptAI.add_predictor( model, @@ -219,7 +219,7 @@ function test_gray_box_scalar_output_hessian() chain = Flux.Chain(Flux.Dense(2 => 16, Flux.relu), Flux.Dense(16 => 1)) model = Model(Ipopt.Optimizer) set_silent(model) - set_attribute(model, "max_iter", 5) + set_attribute(model, "max_iter", 1) @variable(model, 0 <= x[1:2] <= 1) y, formulation = MathOptAI.add_predictor( model, @@ -240,7 +240,7 @@ function test_gray_box_vector_output() chain = Flux.Chain(Flux.Dense(3 => 16, Flux.relu), Flux.Dense(16 => 2)) model = Model(Ipopt.Optimizer) set_silent(model) - set_attribute(model, "max_iter", 5) + set_attribute(model, "max_iter", 1) @variable(model, 0 <= x[1:3] <= 1) y, formulation = MathOptAI.add_predictor( model, @@ -261,7 +261,7 @@ function test_gray_box_vector_output_hessian() chain = Flux.Chain(Flux.Dense(3 => 16, Flux.relu), Flux.Dense(16 => 2)) model = Model(Ipopt.Optimizer) set_silent(model) - set_attribute(model, "max_iter", 5) + set_attribute(model, "max_iter", 1) @variable(model, 0 <= x[1:3] <= 1) y, formulation = MathOptAI.add_predictor( model,