From 31baeed614bf7f65aafde545f20a95e84cd293b4 Mon Sep 17 00:00:00 2001 From: LysandreJik Date: Thu, 9 Apr 2020 09:09:00 -0400 Subject: [PATCH] Update quotes cc @julien-c --- tests/test_tokenization_common.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tests/test_tokenization_common.py b/tests/test_tokenization_common.py index 924faef458..9f28e01169 100644 --- a/tests/test_tokenization_common.py +++ b/tests/test_tokenization_common.py @@ -25,9 +25,12 @@ from tests.utils import require_tf, require_torch def merge_model_tokenizer_mappings( - model_mapping: "Dict[PretrainedConfig, Union[PreTrainedModel, TFPreTrainedModel]]", # noqa: F821 - tokenizer_mapping: "Dict[PretrainedConfig, Tuple[PreTrainedTokenizer, PreTrainedTokenizerFast]]", # noqa: F821 -) -> "Dict[Union[PreTrainedTokenizer, PreTrainedTokenizerFast], Tuple[PretrainedConfig, Union[PreTrainedModel, TFPreTrainedModel]]]": # noqa: F821 + model_mapping: Dict["PretrainedConfig", Union["PreTrainedModel", "TFPreTrainedModel"]], + tokenizer_mapping: Dict["PretrainedConfig", Tuple["PreTrainedTokenizer", "PreTrainedTokenizerFast"]], +) -> Dict[ + Union["PreTrainedTokenizer", "PreTrainedTokenizerFast"], + Tuple["PretrainedConfig", Union["PreTrainedModel", "TFPreTrainedModel"]], +]: configurations = list(model_mapping.keys()) model_tokenizer_mapping = OrderedDict([])