diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-11 23:09:22 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-11 23:09:22 +0200 |
| commit | 6df941bdf5cad80db38d851dcb23a08a9dc55617 (patch) | |
| tree | ecbe41259990955201f910f0113c723cd080ab78 /text_recognizer/data/iam_extended_paragraphs.py | |
| parent | 8177b5210fbbe11bd361dedbd5b4a4e1950bdb2e (diff) | |
Fix emnist mapping
Diffstat (limited to 'text_recognizer/data/iam_extended_paragraphs.py')
| -rw-r--r-- | text_recognizer/data/iam_extended_paragraphs.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/text_recognizer/data/iam_extended_paragraphs.py b/text_recognizer/data/iam_extended_paragraphs.py index 61bf6a3..90df5f8 100644 --- a/text_recognizer/data/iam_extended_paragraphs.py +++ b/text_recognizer/data/iam_extended_paragraphs.py @@ -1,5 +1,5 @@ """IAM original and sythetic dataset class.""" -from typing import Callable, Optional, Type +from typing import Callable, Optional from torch.utils.data import ConcatDataset from text_recognizer.data.base_data_module import BaseDataModule, load_and_print_info @@ -14,7 +14,7 @@ class IAMExtendedParagraphs(BaseDataModule): def __init__( self, - mapping: Type[EmnistMapping], + mapping: EmnistMapping, transform: Optional[Callable] = None, test_transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, |