From 7dfa030557ad500a54126b8af40f6f0c42c5d620 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Thu, 16 Jun 2022 20:25:42 +0200 Subject: Set defaults --- text_recognizer/data/base_data_module.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 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 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: -- cgit v1.2.3-70-g09d2