summaryrefslogtreecommitdiff
path: root/text_recognizer/data
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/data')
-rw-r--r--text_recognizer/data/base_data_module.py10
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: