frozenleaves/trl
1
0
Fork 0
mirror of https://github.com/huggingface/trl.git synced 2025-10-20 18:43:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix CI slow test AttributeError: 'TestSFTTrainerSlow' object has no attribute 'addCleanup' (#4255)

Browse Source
This commit is contained in:
Albert Villanova del Moral
2025-10-10 17:19:53 +02:00
committed by GitHub
parent 039d526d24
commit 86d1963cc1
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/slow/test_sft_slow.py
View File

@ -412,12 +412,12 @@ class TestSFTTrainerSlow(TrlTestCase):
eval_dataset=self.eval_dataset, eval_dataset=self.eval_dataset,
) )
# Register cleanup now that we have the trainer # Ensure cleanup of liger patches after the test
self.addCleanup(cleanup_liger_patches, trainer) try:
trainer.train()
trainer.train() release_memory(trainer.model, trainer)
finally:
release_memory(trainer.model, trainer) cleanup_liger_patches(trainer)
@parameterized.expand(list(itertools.product(MODELS_TO_TEST, PACKING_OPTIONS))) @parameterized.expand(list(itertools.product(MODELS_TO_TEST, PACKING_OPTIONS)))
@require_torch_accelerator @require_torch_accelerator