Expand test to locate flakiness (#7580)

d3adb985 · Sylvain Gugger · GitHub · b2b7fc78 · d3adb985
Unverified Commit d3adb985 authored Oct 05, 2020 by Sylvain Gugger Committed by GitHub Oct 05, 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
@@ -193,7 +193,9 @@ class TrainerIntegrationTest(unittest.TestCase):
        trainer = get_regression_trainer()
        trainer.train()
        args = TrainingArguments("./regression")
-        self.assertEqual(args.to_dict(), trainer.args.to_dict())
+        dict1, dict2 = args.to_dict(), trainer.args.to_dict()
+        for key in dict1.keys():
+            self.assertEqual(dict1[key], dict2[key])

    def test_reproducible_training(self):
        # Checks that training worked, model trained and seed made a reproducible training.