Fix flaky test in test_trainer (#7689)

c6e18de9 · Sylvain Gugger · GitHub · 2c9e83f7 · c6e18de9
Unverified Commit c6e18de9 authored Oct 09, 2020 by Sylvain Gugger Committed by GitHub Oct 09, 2020
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

tests/test_trainer.py tests/test_trainer.py +3 -1

No files found.
--- a/tests/test_trainer.py
+++ b/tests/test_trainer.py
@@ -195,6 +195,8 @@ class TrainerIntegrationTest(unittest.TestCase):
        args = TrainingArguments("./regression")
        dict1, dict2 = args.to_dict(), trainer.args.to_dict()
        for key in dict1.keys():
+            # Logging dir can be slightly different as they default to something with the time.
+            if key != "loggin_dir":
                self.assertEqual(dict1[key], dict2[key])

    def test_reproducible_training(self):