SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[logging] implement warning_advice / TRANSFORMERS_NO_ADVISORY_WARNINGS (#14669)

Browse Source 
* [logging] implement warning_advice / TRANSFORMERS_NO_ADVISORY_WARNINGS

* reword
This commit is contained in:
Stas Bekman
2021-12-20 20:48:38 -08:00
committed by GitHub
parent c1125dc2ba
commit b6ec956976
4 changed files with 44 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/test_logging.py
View File

@@ -17,7 +17,7 @@ import unittest
import transformers.models.bart.tokenization_bart
from transformers import logging
from transformers.testing_utils import CaptureLogger, mockenv
from transformers.testing_utils import CaptureLogger, mockenv, mockenv_context
class HfArgumentParserTest(unittest.TestCase):
@@ -103,3 +103,21 @@ class HfArgumentParserTest(unittest.TestCase):
self.assertIn("Unknown option TRANSFORMERS_VERBOSITY=super-error", cl.out)
# no need to restore as nothing was changed
def test_advisory_warnings(self):
# testing `logger.warning_advice()`
logger = logging.get_logger("transformers.models.bart.tokenization_bart")
msg = "Testing 1, 2, 3"
with mockenv_context(TRANSFORMERS_NO_ADVISORY_WARNINGS="1"):
# nothing should be logged as env var disables this method
with CaptureLogger(logger) as cl:
logger.warning_advice(msg)
self.assertEqual(cl.out, "")
with mockenv_context(TRANSFORMERS_NO_ADVISORY_WARNINGS=""):
# should log normally as TRANSFORMERS_NO_ADVISORY_WARNINGS is unset
with CaptureLogger(logger) as cl:
logger.warning_advice(msg)
self.assertEqual(cl.out, msg + "\n")