mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-13 01:30:04 +06:00
77abd1e79f
* Logging * Style * hf_logging > utils.logging * Address @thomwolf's comments * Update test * Update src/transformers/benchmark/benchmark_utils.py Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> * Revert bad change Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
21 lines
636 B
Python
21 lines
636 B
Python
import unittest
|
|
|
|
from transformers import logging
|
|
|
|
|
|
class HfArgumentParserTest(unittest.TestCase):
|
|
def test_set_level(self):
|
|
logger = logging.get_logger()
|
|
|
|
logging.set_verbosity_error()
|
|
self.assertEqual(logger.getEffectiveLevel(), logging.get_verbosity())
|
|
|
|
logging.set_verbosity_warning()
|
|
self.assertEqual(logger.getEffectiveLevel(), logging.get_verbosity())
|
|
|
|
logging.set_verbosity_info()
|
|
self.assertEqual(logger.getEffectiveLevel(), logging.get_verbosity())
|
|
|
|
logging.set_verbosity_debug()
|
|
self.assertEqual(logger.getEffectiveLevel(), logging.get_verbosity())
|