Tweak typing for #3566

This commit is contained in:
Julien Chaumond
2020-04-04 15:04:03 -04:00
parent 94eb68d742
commit 5d912e7ed4

View File

@@ -19,6 +19,7 @@ import collections
import logging import logging
import os import os
import unicodedata import unicodedata
from typing import Optional
from .tokenization_bert import BasicTokenizer, BertTokenizer, WordpieceTokenizer, load_vocab from .tokenization_bert import BasicTokenizer, BertTokenizer, WordpieceTokenizer, load_vocab
@@ -165,10 +166,10 @@ class BertJapaneseTokenizer(BertTokenizer):
return split_tokens return split_tokens
class MecabTokenizer(object): class MecabTokenizer:
"""Runs basic tokenization with MeCab morphological parser.""" """Runs basic tokenization with MeCab morphological parser."""
def __init__(self, do_lower_case=False, never_split=None, normalize_text=True, mecab_option=None): def __init__(self, do_lower_case=False, never_split=None, normalize_text=True, mecab_option: Optional[str] = None):
"""Constructs a MecabTokenizer. """Constructs a MecabTokenizer.
Args: Args: