update: test_rex_lr_scheduler

kozistr · kozistr · commit a3b1bfb1c7a1 · 2024-03-02T18:34:39.000+09:00
diff --git a/tests/test_lr_schedulers.py b/tests/test_lr_schedulers.py
@@ -283,8 +283,8 @@ def test_rex_lr_scheduler():
     )
 
     for expected_lr in lrs:
-        lr: float = lr_scheduler.step()
-        np.testing.assert_almost_equal(expected_lr, lr, 6)
+        _ = lr_scheduler.step()
+        np.testing.assert_almost_equal(expected_lr, lr_scheduler.get_lr(), 6)
 
 
 def test_deberta_v3_large_lr_scheduler():

Original file line number	Diff line number	Diff line change
`@@ -283,8 +283,8 @@ def test_rex_lr_scheduler():`
`283`	`283`	`)`
`284`	`284`
`285`	`285`	`for expected_lr in lrs:`
`286`		`- lr: float = lr_scheduler.step()`
`287`		`- np.testing.assert_almost_equal(expected_lr, lr, 6)`
	`286`	`+ _ = lr_scheduler.step()`
	`287`	`+ np.testing.assert_almost_equal(expected_lr, lr_scheduler.get_lr(), 6)`
`288`	`288`
`289`	`289`
`290`	`290`	`def test_deberta_v3_large_lr_scheduler():`