CI: fix test-save-trainer (#36191)

* fix * also the docstring
2025-02-14 10:20:56 +01:00
parent 0ae93d31ce
commit 3bf02cf440
1 changed files with 2 additions and 2 deletions
--- 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."""