more explicit import

Yoctol · Jan 24, 2019 · 2d91ec6 · 2d91ec6
1 parent 56a5ef0
commit 2d91ec6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/uttut/pipeline/bert/tests/test_basic.py b/uttut/pipeline/bert/tests/test_basic.py
@@ -1,6 +1,6 @@
 import pytest
 
-from .tokenization import BasicTokenizer
+from .tokenization import BasicTokenizer as BertBasicTokenizer
 from ..basic import basic_pipe
 
 from uttut.elements import Datum
@@ -28,7 +28,7 @@
 
 @pytest.fixture
 def tokenizer():
-    yield BasicTokenizer()
+    yield BertBasicTokenizer()
 
 
 @pytest.mark.parametrize("input_str", test_cases)

diff --git a/uttut/pipeline/bert/tests/test_full.py b/uttut/pipeline/bert/tests/test_full.py
@@ -4,7 +4,7 @@
 
 import pytest
 
-from .tokenization import FullTokenizer
+from .tokenization import FullTokenizer as BertFullTokenizer
 from ..full import full_pipe, vocab_tokens
 from uttut.elements import Datum
 
@@ -28,7 +28,7 @@ def tokenizer():
 
         vocab_file = vocab_writer.name
 
-    tokenizer = FullTokenizer(vocab_file)
+    tokenizer = BertFullTokenizer(vocab_file)
     yield tokenizer
     os.unlink(vocab_file)
 

diff --git a/uttut/pipeline/bert/tests/test_word_piece.py b/uttut/pipeline/bert/tests/test_word_piece.py
@@ -1,6 +1,6 @@
 import pytest
 
-from .tokenization import WordpieceTokenizer
+from .tokenization import WordpieceTokenizer as BertWordpieceTokenizer
 from ..word_piece import word_piece_pipe, vocab
 
 from uttut.elements import Datum
@@ -20,7 +20,7 @@
 
 @pytest.fixture
 def tokenizer():
-    yield WordpieceTokenizer(vocab)
+    yield BertWordpieceTokenizer(vocab)
 
 
 @pytest.mark.parametrize("input_str", test_cases)