Skip to content

Commit 6c2ea00

Browse files
fix!: import WordPiece only if HAS_HF true to avoid general import bug
1 parent c9382ca commit 6c2ea00

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

torchTextClassifiers/tokenizers/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,4 +7,6 @@
77
)
88
from .base import TokenizerOutput as TokenizerOutput
99
from .ngram import NGramTokenizer as NGramTokenizer
10-
from .WordPiece import WordPieceTokenizer as WordPieceTokenizer
10+
11+
if HAS_HF:
12+
from .WordPiece import WordPieceTokenizer as WordPieceTokenizer

0 commit comments

Comments
 (0)