SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 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
1 changed files with 39 additions and 38 deletions
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])