From 2d4714fcfeb8914f240a0d36d938b434e82f191b Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 4 Apr 2021 23:08:16 +0200
Subject: Add new transformer network

---
 text_recognizer/data/emnist.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'text_recognizer/data/emnist.py')

diff --git a/text_recognizer/data/emnist.py b/text_recognizer/data/emnist.py
index eda490a..12adaab 100644
--- a/text_recognizer/data/emnist.py
+++ b/text_recognizer/data/emnist.py
@@ -96,7 +96,7 @@ class EMNIST(BaseDataModule):
 
 
 def emnist_mapping(
-    extra_symbols: Optional[List[str]],
+    extra_symbols: Optional[Sequence[str]],
 ) -> Tuple[List, Dict[str, int], List[int]]:
     """Return the EMNIST mapping."""
     if not ESSENTIALS_FILENAME.exists():
@@ -209,7 +209,7 @@ def _augment_emnist_characters(characters: Sequence[str]) -> Sequence[str]:
     # - End token at index 2
     # - Padding token at index 3
     # Note: Do not forget to update NUM_SPECIAL_TOKENS if changing this!
-    return ["<b>", "<s>", "</s>", "<p>", *characters, *iam_characters]
+    return ["<b>", "<s>", "<e>", "<p>", *characters, *iam_characters]
 
 
 def download_emnist() -> None:
-- 
cgit v1.2.3-70-g09d2