riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-24 14:58:56 +06:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

Modify warnings in a with block to avoid flaky tests (#31893)

Browse Source 
* fix

* [test_all] check before merge

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2024-07-10 17:56:12 +02:00 committed by GitHub
parent ec03d97b27
commit 080e14b24c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 39 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/trainer/test_trainer_callback.py
View File

@ -218,6 +218,7 @@ class TrainerCallbackTest(unittest.TestCase):
import warnings import warnings
# XXX: for now ignore scatter_gather warnings in this test since it's not relevant to what's being tested # XXX: for now ignore scatter_gather warnings in this test since it's not relevant to what's being tested
with warnings.catch_warnings():
warnings.simplefilter(action="ignore", category=UserWarning) warnings.simplefilter(action="ignore", category=UserWarning)
trainer = self.get_trainer(callbacks=[MyTestTrainerCallback]) trainer = self.get_trainer(callbacks=[MyTestTrainerCallback])