diff --git a/tests/test_training_args.py b/tests/test_training_args.py index 7b1daabe16..a4da834582 100644 --- a/tests/test_training_args.py +++ b/tests/test_training_args.py @@ -7,9 +7,9 @@ from transformers import TrainingArguments class TestTrainingArguments(unittest.TestCase): def test_default_output_dir(self): - """Test that output_dir defaults to 'tmp_trainer' when not specified.""" + """Test that output_dir defaults to 'trainer_output' when not specified.""" args = TrainingArguments(output_dir=None) - self.assertEqual(args.output_dir, "tmp_trainer") + self.assertEqual(args.output_dir, "trainer_output") def test_custom_output_dir(self): """Test that output_dir is respected when specified."""