From 2b63fd952bdc9c7c72edd501cbcdbf3231e98f00 Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Wed, 9 Sep 2020 23:31:31 +0200
Subject: Created an abstract Dataset class for common methods.

---
 src/text_recognizer/datasets/__init__.py | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

(limited to 'src/text_recognizer/datasets/__init__.py')

diff --git a/src/text_recognizer/datasets/__init__.py b/src/text_recognizer/datasets/__init__.py
index ede4541..a3af9b1 100644
--- a/src/text_recognizer/datasets/__init__.py
+++ b/src/text_recognizer/datasets/__init__.py
@@ -1,10 +1,5 @@
 """Dataset modules."""
-from .emnist_dataset import (
-    DATA_DIRNAME,
-    EmnistDataset,
-    EmnistMapper,
-    ESSENTIALS_FILENAME,
-)
+from .emnist_dataset import EmnistDataset, Transpose
 from .emnist_lines_dataset import (
     construct_image_from_string,
     EmnistLinesDataset,
@@ -13,7 +8,14 @@ from .emnist_lines_dataset import (
 from .iam_dataset import IamDataset
 from .iam_lines_dataset import IamLinesDataset
 from .iam_paragraphs_dataset import IamParagraphsDataset
-from .util import _download_raw_dataset, compute_sha256, download_url, Transpose
+from .util import (
+    _download_raw_dataset,
+    compute_sha256,
+    DATA_DIRNAME,
+    download_url,
+    EmnistMapper,
+    ESSENTIALS_FILENAME,
+)
 
 __all__ = [
     "_download_raw_dataset",
-- 
cgit v1.2.3-70-g09d2