test: fix flux tests

SciML · Feb 5, 2025 · c04adbd · c04adbd
1 parent c5cae95
commit c04adbd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lib/SurrogatesFlux/test/runtests.jl b/lib/SurrogatesFlux/test/runtests.jl
@@ -53,6 +53,7 @@ using SafeTestsets
         x = sample(20, lb, ub, SobolSample())
         y = f.(x)
         my_model = Chain(Dense(2, 2))
+        my_opt = Descent(0.01)
         surrogate = NeuralSurrogate(x, y, lb, ub, model = my_model, loss = Flux.mse,
             opt = my_opt, n_epochs = 1)
         surrogate([1.0, 2.0])