From 3bf02cf44079c84d5d451ae152f6554d1e1de318 Mon Sep 17 00:00:00 2001 From: Raushan Turganbay Date: Fri, 14 Feb 2025 10:20:56 +0100 Subject: [PATCH] CI: fix `test-save-trainer` (#36191) * fix * also the docstring --- tests/test_training_args.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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."""