comment

huggingface · Jul 18, 2024 · ff09978 · ff09978
1 parent 24834dc
commit ff09978
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
@@ -3791,6 +3791,7 @@ def test_push_to_hub_with_saves_each_n_steps(self):
         max_steps = math.ceil(trainer.args.num_train_epochs * len(trainer.get_train_dataloader()))
         nb_expected_commits = len(range(5, max_steps, 5))
 
+        # '>=' since final commit might be an empty commit as well (not deterministic)
         self.assertGreaterEqual(nb_empty_commits + nb_epoch_commits, nb_expected_commits)
 
     @require_tensorboard