From a8ec52efc217474ff164461bebcfec060cff6837 Mon Sep 17 00:00:00 2001
From: Sylvain Gugger <sylvain.gugger@gmail.com>
Date: Mon, 8 Mar 2021 16:04:46 -0500
Subject: [PATCH] Style

---
 tests/test_trainer.py       | 2 +-
 tests/test_trainer_utils.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_trainer.py b/tests/test_trainer.py
index f29a8a60fc..ae5fe2a819 100644
--- a/tests/test_trainer.py
+++ b/tests/test_trainer.py
@@ -992,7 +992,7 @@ class TrainerIntegrationTest(unittest.TestCase):
         # should be about half of fp16_init
         # perfect world: fp32_init/2 == fp16_eval
         self.assertAlmostEqual(fp16_eval, fp32_init / 2, delta=5_000)
-    
+
     def test_no_wd_param_group(self):
         model = torch.nn.Sequential(TstLayer(128), torch.nn.ModuleList([TstLayer(128), TstLayer(128)]))
         trainer = Trainer(model=model)
diff --git a/tests/test_trainer_utils.py b/tests/test_trainer_utils.py
index 2d9d1d688f..f56ef140e8 100644
--- a/tests/test_trainer_utils.py
+++ b/tests/test_trainer_utils.py
@@ -30,7 +30,7 @@ if is_torch_available():
         DistributedTensorGatherer,
         LabelSmoother,
         LengthGroupedSampler,
-        get_parameter_names
+        get_parameter_names,
     )
 
     class TstLayer(torch.nn.Module):