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

* 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

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])