diff options
Diffstat (limited to 'text_recognizer/data/emnist.py')
| -rw-r--r-- | text_recognizer/data/emnist.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/text_recognizer/data/emnist.py b/text_recognizer/data/emnist.py index dc8d31a..94882bf 100644 --- a/text_recognizer/data/emnist.py +++ b/text_recognizer/data/emnist.py @@ -6,7 +6,7 @@ import shutil from typing import Dict, List, Optional, Sequence, Set, Tuple import zipfile -import attr +from attrs import define import h5py from loguru import logger as log import numpy as np @@ -35,7 +35,7 @@ ESSENTIALS_FILENAME = ( ) -@attr.s(auto_attribs=True) +@define(auto_attribs=True) class EMNIST(BaseDataModule): """Lightning DataModule class for loading EMNIST dataset. |