diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-12-05 20:24:53 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-12-05 20:24:53 +0100 |
| commit | c18a18b803e34687e541d214c8d67e72e3d0c3cb (patch) | |
| tree | faca7e5d6135cd0a2378b6a2505162d7b4689869 /text_recognizer/data/transforms | |
| parent | 95d78c2bc2f6c5851cd1124b109077a04b526f10 (diff) | |
Format files with black
Diffstat (limited to 'text_recognizer/data/transforms')
| -rw-r--r-- | text_recognizer/data/transforms/word_piece.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/text_recognizer/data/transforms/word_piece.py b/text_recognizer/data/transforms/word_piece.py index 69f0ce1..a15615d 100644 --- a/text_recognizer/data/transforms/word_piece.py +++ b/text_recognizer/data/transforms/word_piece.py @@ -19,7 +19,9 @@ class WordPiece: use_words: bool = False, prepend_wordsep: bool = False, special_tokens: Set[str] = {"<s>", "<e>", "<p>"}, - extra_symbols: Optional[Set[str]] = {"\n",}, + extra_symbols: Optional[Set[str]] = { + "\n", + }, max_len: int = 451, ) -> None: self.mapping = WordPieceMapping( |