diff options
| -rw-r--r-- | text_recognizer/data/base_data_module.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/text_recognizer/data/base_data_module.py b/text_recognizer/data/base_data_module.py index 69581a3..7863333 100644 --- a/text_recognizer/data/base_data_module.py +++ b/text_recognizer/data/base_data_module.py @@ -44,11 +44,11 @@ class BaseDataModule(LightningDataModule): self.pin_memory = pin_memory # Placeholders - self.data_train: BaseDataset - self.data_val: BaseDataset - self.data_test: BaseDataset - self.dims: Tuple[int, ...] - self.output_dims: Tuple[int, ...] + self.data_train: BaseDataset = None + self.data_val: BaseDataset = None + self.data_test: BaseDataset = None + self.dims: Tuple[int, ...] = None + self.output_dims: Tuple[int, ...] = None @classmethod def data_dirname(cls: T) -> Path: |