Fix flaky test for log level (#21776)

* Fix flaky test for log level

* Fix other flaky test
This commit is contained in:
Sylvain Gugger
2023-02-28 22:24:14 +01:00
committed by GitHub
parent acfb714bdf
commit b29e2dcaff
2 changed files with 7 additions and 4 deletions

View File

@@ -109,6 +109,7 @@ class HfArgumentParserTest(unittest.TestCase):
def test_advisory_warnings(self):
# testing `logger.warning_advice()`
transformers.utils.logging._reset_library_root_logger()
logger = logging.get_logger("transformers.models.bart.tokenization_bart")
msg = "Testing 1, 2, 3"