Skip to content

Commit 9a567bf

Browse files
committed
fix float error in early stop test
1 parent a19d64c commit 9a567bf

1 file changed

Lines changed: 37 additions & 26 deletions

File tree

tests/test_api/test_early_stop.py

Lines changed: 37 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -102,25 +102,28 @@ def objective_function(para):
102102

103103

104104
def test_early_stop_4():
105-
def objective_function(para):
106-
return para["x0"]
105+
class Experiment(BaseExperiment):
106+
def objective_function(self, para):
107+
return para["x0"]
107108

108-
search_space = {
109-
"x0": list(np.arange(0, 100, 0.1)),
110-
}
109+
experiment = Experiment()
110+
111+
search_config = SearchConfig(
112+
x0=list(np.arange(0, 100, 0.01)),
113+
)
111114

112115
n_iter_no_change = 5
113116
early_stopping = {
114117
"n_iter_no_change": 5,
115-
"tol_abs": 0.1,
118+
"tol_abs": 1,
116119
"tol_rel": None,
117120
}
118121

119122
start1 = {"x0": 0}
120-
start2 = {"x0": 0.1}
121-
start3 = {"x0": 0.2}
122-
start4 = {"x0": 0.3}
123-
start5 = {"x0": 0.4}
123+
start2 = {"x0": 1}
124+
start3 = {"x0": 2}
125+
start4 = {"x0": 3}
126+
start5 = {"x0": 4}
124127

125128
warm_start_l = [
126129
start1,
@@ -163,23 +166,26 @@ def objective_function(para):
163166

164167

165168
def test_early_stop_5():
166-
def objective_function(para):
167-
return para["x0"]
169+
class Experiment(BaseExperiment):
170+
def objective_function(self, para):
171+
return para["x0"]
168172

169-
search_space = {
170-
"x0": list(np.arange(0, 100, 0.01)),
171-
}
173+
experiment = Experiment()
174+
175+
search_config = SearchConfig(
176+
x0=list(np.arange(0, 100, 0.01)),
177+
)
172178

173179
n_iter_no_change = 5
174180
early_stopping = {
175181
"n_iter_no_change": n_iter_no_change,
176-
"tol_abs": 0.1,
182+
"tol_abs": 10,
177183
"tol_rel": None,
178184
}
179185

180186
start1 = {"x0": 0}
181-
start2 = {"x0": 0.09}
182-
start3 = {"x0": 0.20}
187+
start2 = {"x0": 9}
188+
start3 = {"x0": 20}
183189

184190
warm_start_l = [
185191
start1,
@@ -216,12 +222,15 @@ def objective_function(para):
216222

217223

218224
def test_early_stop_6():
219-
def objective_function(para):
220-
return para["x0"]
225+
class Experiment(BaseExperiment):
226+
def objective_function(self, para):
227+
return para["x0"]
221228

222-
search_space = {
223-
"x0": list(np.arange(0, 100, 0.01)),
224-
}
229+
experiment = Experiment()
230+
231+
search_config = SearchConfig(
232+
x0=list(np.arange(0, 100, 0.01)),
233+
)
225234

226235
n_iter_no_change = 5
227236
early_stopping = {
@@ -280,9 +289,11 @@ def test_early_stop_7():
280289
def objective_function(para):
281290
return para["x0"]
282291

283-
search_space = {
284-
"x0": list(np.arange(0, 100, 0.01)),
285-
}
292+
experiment = Experiment()
293+
294+
search_config = SearchConfig(
295+
x0=list(np.arange(0, 100, 0.01)),
296+
)
286297

287298
n_iter_no_change = 5
288299
early_stopping = {

0 commit comments

Comments
 (0)