diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-05 23:39:11 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-05 23:39:11 +0200 |
| commit | 65df6a72b002c4b23d6f2eb545839e157f7f2aa0 (patch) | |
| tree | d78df1d7143dc9ff9e29afd4fd6bc7490bc79418 /text_recognizer/data/emnist.py | |
| parent | 8bc4b4cab00a2777a748c10fca9b3ee01e32277c (diff) | |
Remove attrs
Diffstat (limited to 'text_recognizer/data/emnist.py')
| -rw-r--r-- | text_recognizer/data/emnist.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/text_recognizer/data/emnist.py b/text_recognizer/data/emnist.py index 94882bf..1b1381a 100644 --- a/text_recognizer/data/emnist.py +++ b/text_recognizer/data/emnist.py @@ -6,7 +6,6 @@ import shutil from typing import Dict, List, Optional, Sequence, Set, Tuple import zipfile -from attrs import define import h5py from loguru import logger as log import numpy as np @@ -35,7 +34,6 @@ ESSENTIALS_FILENAME = ( ) -@define(auto_attribs=True) class EMNIST(BaseDataModule): """Lightning DataModule class for loading EMNIST dataset. @@ -48,8 +46,8 @@ class EMNIST(BaseDataModule): EMNIST ByClass: 814,255 characters. 62 unbalanced classes. """ - def __attrs_post_init__(self) -> None: - """Post init configuration.""" + def __init__(self) -> None: + super().__init__() self.dims = (1, *self.mapping.input_size) def prepare_data(self) -> None: |