Keep basic optim test LR range closer to before w/ updated code

huggingface · Nov 26, 2024 · 1a70036 · 1a70036
1 parent 269bc08
commit 1a70036
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_optim.py b/tests/test_optim.py
@@ -297,9 +297,9 @@ def test_optim_factory(optimizer):
     opt_info = get_optimizer_info(optimizer)
     assert isinstance(opt_info, OptimInfo)
 
-    lr = (1e-3, 1e-2, 1e-2, 1e-2)
+    lr = (1e-2,) * 4
     if optimizer in ('mars',):
-        lr = (1e-3, 1e-3, 1e-3, 1e-3)
+        lr = (1e-3,) * 4
 
     try:
         if not opt_info.second_order:  # basic tests don't support second order right now