From 0d0540952f79437026fc5a146b81e4b45190ff6a Mon Sep 17 00:00:00 2001 From: aktersnurra Date: Mon, 3 Aug 2020 23:56:51 +0200 Subject: Fix import error in datasets package. --- src/text_recognizer/datasets/emnist_lines_dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'src/text_recognizer/datasets/emnist_lines_dataset.py') 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" -- cgit v1.2.3-70-g09d2