Enable tqdm toggling (#15167)
* feature: enable tqdm toggle * test: add tqdm unit test * style: run linter * Update tests/test_tqdm_utils.py Co-authored-by: Stas Bekman <stas00@users.noreply.github.com> * refactor: use tiny model, run linter * docs: add tqdm to logging * docs: add tqdm reference to `http_get` * style: run linter * Update docs/source/main_classes/logging.mdx Co-authored-by: Stas Bekman <stas00@users.noreply.github.com> * fix: use `AutoConfig` for framework agnostic testing * chore: mv tqdm test to `test_logging.py` * feature: implement enable/disable functions * docs: mv docstring to comment * chore: mv tqdm functions to `logging.py` * docs: update docs to reference `enable/disable` funcs * test: update test to use `enable/disable` func * chore: update function reference in comment Co-authored-by: Stas Bekman <stas00@users.noreply.github.com>
This commit is contained in:
@@ -14,10 +14,12 @@
|
||||
|
||||
import os
|
||||
import unittest
|
||||
from unittest.mock import patch
|
||||
|
||||
import transformers.models.bart.tokenization_bart
|
||||
from transformers import logging
|
||||
from transformers import AutoConfig, logging
|
||||
from transformers.testing_utils import CaptureLogger, mockenv, mockenv_context
|
||||
from transformers.utils.logging import disable_progress_bar, enable_progress_bar
|
||||
|
||||
|
||||
class HfArgumentParserTest(unittest.TestCase):
|
||||
@@ -121,3 +123,17 @@ class HfArgumentParserTest(unittest.TestCase):
|
||||
with CaptureLogger(logger) as cl:
|
||||
logger.warning_advice(msg)
|
||||
self.assertEqual(cl.out, msg + "\n")
|
||||
|
||||
|
||||
def test_set_progress_bar_enabled():
|
||||
TINY_MODEL = "hf-internal-testing/tiny-random-distilbert"
|
||||
with patch("tqdm.auto.tqdm") as mock_tqdm:
|
||||
disable_progress_bar()
|
||||
_ = AutoConfig.from_pretrained(TINY_MODEL, force_download=True)
|
||||
mock_tqdm.assert_not_called()
|
||||
|
||||
mock_tqdm.reset_mock()
|
||||
|
||||
enable_progress_bar()
|
||||
_ = AutoConfig.from_pretrained(TINY_MODEL, force_download=True)
|
||||
mock_tqdm.assert_called()
|
||||
|
||||
Reference in New Issue
Block a user