diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-27 23:11:06 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-27 23:11:06 +0200 |
commit | 9c7dbb9ca70858b870f74ecf595d3169f0cbc711 (patch) | |
tree | c342e2c004bb75571a380ef2805049a8fcec3fcc /text_recognizer/data/base_data_module.py | |
parent | 9b8e14d89f0ef2508ed11f994f73af624155fe1d (diff) |
Rename mapping to tokenizer
Diffstat (limited to 'text_recognizer/data/base_data_module.py')
-rw-r--r-- | text_recognizer/data/base_data_module.py | 11 |
1 files changed, 3 insertions, 8 deletions
diff --git a/text_recognizer/data/base_data_module.py b/text_recognizer/data/base_data_module.py index 7863333..bd6fd99 100644 --- a/text_recognizer/data/base_data_module.py +++ b/text_recognizer/data/base_data_module.py @@ -6,7 +6,7 @@ from pytorch_lightning import LightningDataModule from torch.utils.data import DataLoader from text_recognizer.data.base_dataset import BaseDataset -from text_recognizer.data.mappings import EmnistMapping +from text_recognizer.data.tokenizer import Tokenizer T = TypeVar("T") @@ -24,7 +24,7 @@ class BaseDataModule(LightningDataModule): def __init__( self, - mapping: EmnistMapping, + tokenizer: Tokenizer, transform: Optional[Callable] = None, test_transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, @@ -34,7 +34,7 @@ class BaseDataModule(LightningDataModule): pin_memory: bool = True, ) -> None: super().__init__() - self.mapping = mapping + self.tokenizer = tokenizer self.transform = transform self.test_transform = test_transform self.target_transform = target_transform @@ -50,11 +50,6 @@ class BaseDataModule(LightningDataModule): self.dims: Tuple[int, ...] = None self.output_dims: Tuple[int, ...] = None - @classmethod - def data_dirname(cls: T) -> Path: - """Return the path to the base data directory.""" - return Path(__file__).resolve().parents[2] / "data" - def config(self) -> Dict: """Return important settings of the dataset.""" return { |