Skip to content

Commit e343055

Browse files
committed
fix test_gradient
1 parent a25c1a7 commit e343055

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_gradient.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ ApplicationDrivenLearning.set_forecast_model(model, Chain(Dense(1 => 1)))
1414

1515
@testset "GradientMode Stop Rules" begin
1616
# epochs
17+
initial_sol = ApplicationDrivenLearning.extract_params(model.forecast)
1718
opt = ApplicationDrivenLearning.Options(
1819
ApplicationDrivenLearning.GradientMode,
1920
epochs = 0,
@@ -22,7 +23,6 @@ ApplicationDrivenLearning.set_forecast_model(model, Chain(Dense(1 => 1)))
2223
@test initial_sol == sol.params
2324

2425
# time_limit
25-
initial_sol = ApplicationDrivenLearning.extract_params(model.forecast)
2626
opt = ApplicationDrivenLearning.Options(
2727
ApplicationDrivenLearning.GradientMode,
2828
time_limit = 0,

0 commit comments

Comments
 (0)