diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-16 20:25:42 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-16 20:25:42 +0200 |
| commit | 7dfa030557ad500a54126b8af40f6f0c42c5d620 (patch) | |
| tree | 8cca6764fb1d08935b0a2f048462f62e4f52a1b7 | |
| parent | f616bb5e67361ac17b8a153a45863d02b788c861 (diff) | |
Set defaults
| -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: |