From 42fddacd1cac3cc57c3326aa51a409f5090b1261 Mon Sep 17 00:00:00 2001 From: Lysandre Date: Thu, 27 Aug 2020 18:31:51 +0200 Subject: [PATCH] Format --- tests/test_optimization.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/test_optimization.py b/tests/test_optimization.py index 3c79c40208..153b8d1dec 100644 --- 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)