From cb3e5c33f7c2131f723ab79b88f98f8e1df6d927 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Thu, 19 Nov 2020 11:56:42 -0500 Subject: [PATCH] Fix a few last paths for the new repo org (#8666) --- docs/source/testing.rst | 2 +- examples/contrib/run_chinese_ref.py | 2 +- examples/longform-qa/eli5_app.py | 7 ++++++- model_cards/julien-c/dummy-unknown/README.md | 3 +-- src/transformers/testing_utils.py | 2 +- 5 files changed, 10 insertions(+), 6 deletions(-) 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"