From 9c7dbb9ca70858b870f74ecf595d3169f0cbc711 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 27 Sep 2022 23:11:06 +0200 Subject: Rename mapping to tokenizer --- text_recognizer/data/base_data_module.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) (limited to 'text_recognizer/data/base_data_module.py') 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 { -- cgit v1.2.3-70-g09d2