summaryrefslogtreecommitdiff
path: root/text_recognizer/data/emnist_lines.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-11 23:09:22 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-11 23:09:22 +0200
commit6df941bdf5cad80db38d851dcb23a08a9dc55617 (patch)
treeecbe41259990955201f910f0113c723cd080ab78 /text_recognizer/data/emnist_lines.py
parent8177b5210fbbe11bd361dedbd5b4a4e1950bdb2e (diff)
Fix emnist mapping
Diffstat (limited to 'text_recognizer/data/emnist_lines.py')
-rw-r--r--text_recognizer/data/emnist_lines.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/data/emnist_lines.py b/text_recognizer/data/emnist_lines.py
index ba1b61c..2107d74 100644
--- a/text_recognizer/data/emnist_lines.py
+++ b/text_recognizer/data/emnist_lines.py
@@ -38,7 +38,7 @@ class EMNISTLines(BaseDataModule):
def __init__(
self,
- mapping: Type[EmnistMapping],
+ mapping: EmnistMapping,
transform: Optional[Callable] = None,
test_transform: Optional[Callable] = None,
target_transform: Optional[Callable] = None,