Fix #5544 (#5551)

7833b21a · Sylvain Gugger · GitHub · c4734840 · 7833b21a
Unverified Commit 7833b21a authored Jul 06, 2020 by Sylvain Gugger Committed by GitHub Jul 06, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/tokenization_utils.py src/transformers/tokenization_utils.py +1 -1

No files found.
--- a/src/transformers/tokenization_utils.py
+++ b/src/transformers/tokenization_utils.py
@@ -649,7 +649,7 @@ class PreTrainedTokenizer(PreTrainedTokenizerBase):

    def convert_ids_to_tokens(
        self, ids: Union[int, List[int]], skip_special_tokens: bool = False
-    ) -> Union[int, List[int]]:
+    ) -> Union[str, List[str]]:
        """ Converts a single index or a sequence of indices (integers) in a token "
            (resp.) a sequence of tokens (str), using the vocabulary and added tokens.