Skip to content

Commit c04adbd

Browse files
test: fix flux tests
1 parent c5cae95 commit c04adbd

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

lib/SurrogatesFlux/test/runtests.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ using SafeTestsets
5353
x = sample(20, lb, ub, SobolSample())
5454
y = f.(x)
5555
my_model = Chain(Dense(2, 2))
56+
my_opt = Descent(0.01)
5657
surrogate = NeuralSurrogate(x, y, lb, ub, model = my_model, loss = Flux.mse,
5758
opt = my_opt, n_epochs = 1)
5859
surrogate([1.0, 2.0])

0 commit comments

Comments
 (0)