diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-08-03 23:56:51 +0200 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-08-03 23:56:51 +0200 |
| commit | 0d0540952f79437026fc5a146b81e4b45190ff6a (patch) | |
| tree | 05ed905362ac77706b2719643e74479f578ababa /src/text_recognizer/datasets | |
| parent | d570eaaafeced22db3263636e538eddb26b29871 (diff) | |
Fix import error in datasets package.
Diffstat (limited to 'src/text_recognizer/datasets')
| -rw-r--r-- | src/text_recognizer/datasets/__init__.py | 2 | ||||
| -rw-r--r-- | src/text_recognizer/datasets/emnist_lines_dataset.py | 3 |
2 files changed, 2 insertions, 3 deletions
diff --git a/src/text_recognizer/datasets/__init__.py b/src/text_recognizer/datasets/__init__.py index bfa6a02..a8c46c4 100644 --- a/src/text_recognizer/datasets/__init__.py +++ b/src/text_recognizer/datasets/__init__.py @@ -9,7 +9,6 @@ from .emnist_lines_dataset import ( EmnistLinesDataset, get_samples_by_character, ) -from .sentence_generator import SentenceGenerator from .util import Transpose __all__ = [ @@ -19,6 +18,5 @@ __all__ = [ "EmnistDataLoaders", "EmnistLinesDataset", "get_samples_by_character", - "SentenceGenerator", "Transpose", ] diff --git a/src/text_recognizer/datasets/emnist_lines_dataset.py b/src/text_recognizer/datasets/emnist_lines_dataset.py index d49319f..4d8b646 100644 --- a/src/text_recognizer/datasets/emnist_lines_dataset.py +++ b/src/text_recognizer/datasets/emnist_lines_dataset.py @@ -11,7 +11,8 @@ import torch from torch.utils.data import Dataset from torchvision.transforms import Compose, Normalize, ToTensor -from text_recognizer.datasets import DATA_DIRNAME, EmnistDataset, SentenceGenerator +from text_recognizer.datasets import DATA_DIRNAME, EmnistDataset +from text_recognizer.datasets.sentence_generator import SentenceGenerator from text_recognizer.datasets.util import Transpose DATA_DIRNAME = DATA_DIRNAME / "processed" / "emnist_lines" |