diff --git a/docs/source/testing.rst b/docs/source/testing.rst index b16c77ee24..6015a00404 100644 --- a/docs/source/testing.rst +++ b/docs/source/testing.rst @@ -1055,7 +1055,7 @@ If you need to validate the output of a logger, you can use :obj:`CaptureLogger` msg = "Testing 1, 2, 3" logging.set_verbosity_info() - logger = logging.get_logger("transformers.tokenization_bart") + logger = logging.get_logger("transformers.models.bart.tokenization_bart") with CaptureLogger(logger) as cl: logger.info(msg) assert cl.out, msg+"\n" diff --git a/examples/contrib/run_chinese_ref.py b/examples/contrib/run_chinese_ref.py index 8ec7b7bc50..c5c3c8c871 100644 --- a/examples/contrib/run_chinese_ref.py +++ b/examples/contrib/run_chinese_ref.py @@ -3,7 +3,7 @@ import json from typing import List from ltp import LTP -from transformers.tokenization_bert import BertTokenizer +from transformers import BertTokenizer def _is_chinese_char(cp): diff --git a/examples/longform-qa/eli5_app.py b/examples/longform-qa/eli5_app.py index 4bb8de178e..7782d6433b 100644 --- a/examples/longform-qa/eli5_app.py +++ b/examples/longform-qa/eli5_app.py @@ -107,7 +107,12 @@ def make_support(question, source="wiki40b", method="dense", n_results=10): return question_doc, support_list -@st.cache(hash_funcs={torch.Tensor: (lambda _: None), transformers.tokenization_bart.BartTokenizer: (lambda _: None)}) +@st.cache( + hash_funcs={ + torch.Tensor: (lambda _: None), + transformers.models.bart.tokenization_bart.BartTokenizer: (lambda _: None), + } +) def answer_question( question_doc, s2s_model, s2s_tokenizer, min_len=64, max_len=256, sampling=False, n_beams=2, top_p=0.95, temp=0.8 ): diff --git a/model_cards/julien-c/dummy-unknown/README.md b/model_cards/julien-c/dummy-unknown/README.md index 6eb7329883..dd72d529da 100644 --- a/model_cards/julien-c/dummy-unknown/README.md +++ b/model_cards/julien-c/dummy-unknown/README.md @@ -9,8 +9,7 @@ tags: ```python import json import os -from transformers.configuration_roberta import RobertaConfig -from transformers import RobertaForMaskedLM, TFRobertaForMaskedLM +from transformers import RobertaConfig, RobertaForMaskedLM, TFRobertaForMaskedLM DIRNAME = "./dummy-unknown" diff --git a/src/transformers/testing_utils.py b/src/transformers/testing_utils.py index cbce4e9654..03b53c1768 100644 --- a/src/transformers/testing_utils.py +++ b/src/transformers/testing_utils.py @@ -470,7 +470,7 @@ class CaptureLogger: >>> msg = "Testing 1, 2, 3" >>> logging.set_verbosity_info() - >>> logger = logging.get_logger("transformers.tokenization_bart") + >>> logger = logging.get_logger("transformers.models.bart.tokenization_bart") >>> with CaptureLogger(logger) as cl: ... logger.info(msg) >>> assert cl.out, msg+"\n"