Format

42fddacd · Lysandre · 70fccc5c · 42fddacd
Commit 42fddacd authored Aug 27, 2020 by Lysandre
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

tests/test_optimization.py tests/test_optimization.py +4 -1

No files found.
--- a/tests/test_optimization.py
+++ b/tests/test_optimization.py
@@ -154,7 +154,10 @@ class ScheduleInitTest(unittest.TestCase):
            self.assertEqual(len([scheduler.get_lr()[0]]), 1)
            lrs_1 = unwrap_schedule(scheduler, self.num_steps)
            self.assertListAlmostEqual(
-                lrs_1, expected_learning_rates, tol=1e-2, msg=f"failed for {scheduler_func} in normal scheduler",
+                lrs_1,
+                expected_learning_rates,
+                tol=1e-2,
+                msg=f"failed for {scheduler_func} in normal scheduler",
            )

            scheduler = scheduler_func(self.optimizer, **kwargs)