Merge pull request #11 from WorksApplications/fix/pretokenizer

fix import
WorksApplications · Jul 12, 2021 · ced5a47 · ced5a47
2 parents 60ef00f + 7d3e2fc
commit ced5a47
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sudachitra/pretokenizer/japanese_bert_wordpiece_tokenizer.py b/sudachitra/pretokenizer/japanese_bert_wordpiece_tokenizer.py
@@ -24,7 +24,7 @@
 from tokenizers.implementations import BertWordPieceTokenizer
 from tokenizers.implementations.base_tokenizer import BaseTokenizer
 
-from . import CustomPreTokenizer
+from .sudachipy_pretokenizer import CustomPreTokenizer
 
 
 CPT = TypeVar('CPT', bound=CustomPreTokenizer)

diff --git a/sudachitra/pretokenizer/sudachipy_pretokenizer.py b/sudachitra/pretokenizer/sudachipy_pretokenizer.py
@@ -80,7 +80,7 @@ def __init__(
             **kwargs:
                 Sudachi dictionary parameters.
         """
-        SudachipyWordTokenizer.__init__(split_mode=split_mode, dict_type=dict_type, **kwargs)
+        SudachipyWordTokenizer.__init__(self, split_mode=split_mode, dict_type=dict_type, **kwargs)
         self.word_form_type = word_form_type
         self.word_formatter = WORD_FORM_TYPES[self.word_form_type] if self.word_form_type != "surface" else None