Skip to content

Commit aeb1ed7

Browse files
committed
Keep basic optim test LR range closer to before w/ updated code
1 parent 7a165fc commit aeb1ed7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_optim.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -297,9 +297,9 @@ def test_optim_factory(optimizer):
297297
opt_info = get_optimizer_info(optimizer)
298298
assert isinstance(opt_info, OptimInfo)
299299

300-
lr = (1e-3, 1e-2, 1e-2, 1e-2)
300+
lr = (1e-2,) * 4
301301
if optimizer in ('mars',):
302-
lr = (1e-3, 1e-3, 1e-3, 1e-3)
302+
lr = (1e-3,) * 4
303303

304304
try:
305305
if not opt_info.second_order: # basic tests don't support second order right now

0 commit comments

Comments
 (0)