From 1a70036691684e23f08e1287cb4fd60444781589 Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Tue, 26 Nov 2024 13:40:20 -0800
Subject: [PATCH] Keep basic optim test LR range closer to before w/ updated
 code

---
 tests/test_optim.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_optim.py b/tests/test_optim.py
index 1576b0254..d9827ae84 100644
--- 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