From 53a4af4ca22ced165fde14cd0de46e29aab7d80d Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Wed, 26 Jan 2022 23:17:33 +0100
Subject: fix: change Set to Sequence

---
 text_recognizer/data/transforms/word_piece.py | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

(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 a15615d..d805c7e 100644
--- a/text_recognizer/data/transforms/word_piece.py
+++ b/text_recognizer/data/transforms/word_piece.py
@@ -1,6 +1,5 @@
 """Target transform for word pieces."""
-from pathlib import Path
-from typing import Optional, Union, Set
+from typing import Optional, Sequence
 
 import torch
 from torch import Tensor
@@ -18,10 +17,8 @@ class WordPiece:
         lexicon: str = "iamdb_1kwp_lex_1000.txt",
         use_words: bool = False,
         prepend_wordsep: bool = False,
-        special_tokens: Set[str] = {"<s>", "<e>", "<p>"},
-        extra_symbols: Optional[Set[str]] = {
-            "\n",
-        },
+        special_tokens: Sequence[str] = ("<s>", "<e>", "<p>"),
+        extra_symbols: Optional[Sequence[str]] = ("\n",),
         max_len: int = 451,
     ) -> None:
         self.mapping = WordPieceMapping(
-- 
cgit v1.2.3-70-g09d2