diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py index 6708c484eb..f687d2d88e 100644 --- a/tests/trainer/test_trainer.py +++ b/tests/trainer/test_trainer.py @@ -3300,8 +3300,8 @@ class TrainerIntegrationTest(TestCasePlus, TrainerIntegrationCommon): --learning_rate 2e-5 --num_train_epochs 1 --output_dir {tmpdir} - --auto_find_batch_size 0 --report_to none + --auto_find_batch_size 0 """.split() with self.assertRaises(RuntimeError): with patch.object(sys, "argv", testargs):