Fix CI

adelevie · Aug 17, 2020 · 7ca6ab6 · 7ca6ab6
1 parent b732e7e
commit 7ca6ab6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/transformers/tokenization_bart.py b/src/transformers/tokenization_bart.py
@@ -14,8 +14,10 @@
 # limitations under the License.
 
 import logging
+from typing import List, Optional
 
 from .tokenization_roberta import RobertaTokenizer, RobertaTokenizerFast
+from .tokenization_utils_base import BatchEncoding
 
 
 logger = logging.getLogger(__name__)