riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-21 05:28:21 +06:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

Fix auto batch size finder test (#38125)

Browse Source 
Ensure --auto_find_batch_size is the last test arg so indexing is correct
This commit is contained in:
ivarflakstad 2025-05-14 14:12:04 +02:00 committed by GitHub
parent fe918d13b9
commit 1b00966395
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/trainer/test_trainer.py
View File

@ -3300,8 +3300,8 @@ class TrainerIntegrationTest(TestCasePlus, TrainerIntegrationCommon):
--learning_rate 2e-5 --learning_rate 2e-5
--num_train_epochs 1 --num_train_epochs 1
--output_dir {tmpdir} --output_dir {tmpdir}
--auto_find_batch_size 0
--report_to none --report_to none
--auto_find_batch_size 0
""".split() """.split()
with self.assertRaises(RuntimeError): with self.assertRaises(RuntimeError):
with patch.object(sys, "argv", testargs): with patch.object(sys, "argv", testargs):