summaryrefslogtreecommitdiff
path: root/text_recognizer
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-16 20:25:42 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-16 20:25:42 +0200
commit7dfa030557ad500a54126b8af40f6f0c42c5d620 (patch)
tree8cca6764fb1d08935b0a2f048462f62e4f52a1b7 /text_recognizer
parentf616bb5e67361ac17b8a153a45863d02b788c861 (diff)
Set defaults
Diffstat (limited to 'text_recognizer')
-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: