From c18a18b803e34687e541d214c8d67e72e3d0c3cb Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 5 Dec 2021 20:24:53 +0100 Subject: Format files with black --- text_recognizer/data/transforms/word_piece.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'text_recognizer/data/transforms') 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] = {"", "", "

"}, - extra_symbols: Optional[Set[str]] = {"\n",}, + extra_symbols: Optional[Set[str]] = { + "\n", + }, max_len: int = 451, ) -> None: self.mapping = WordPieceMapping( -- cgit v1.2.3-70-g09d2