SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source 
* fix

* also the docstring
This commit is contained in:
Raushan Turganbay
2025-02-14 10:20:56 +01:00
committed by GitHub
parent 0ae93d31ce
commit 3bf02cf440
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_training_args.py
View File

@@ -7,9 +7,9 @@ from transformers import TrainingArguments
class TestTrainingArguments(unittest.TestCase): class TestTrainingArguments(unittest.TestCase):
def test_default_output_dir(self): 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) 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): def test_custom_output_dir(self):
"""Test that output_dir is respected when specified.""" """Test that output_dir is respected when specified."""