Format

Narsil · Aug 27, 2020 · 42fddac · 42fddac
1 parent 70fccc5
commit 42fddac
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_optimization.py b/tests/test_optimization.py
@@ -154,7 +154,10 @@ def test_schedulers(self):
             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)