1 files changed, 22 insertions, 2 deletions
diff --git a/src/text_recognizer/datasets/__init__.py b/src/text_recognizer/datasets/__init__.py
index 795be90..bfa6a02 100644
--- a/src/text_recognizer/datasets/__init__.py
+++ b/src/text_recognizer/datasets/__init__.py
@@ -1,4 +1,24 @@
 """Dataset modules."""
-from .emnist_dataset import EmnistDataLoader
+from .emnist_dataset import (
+    DATA_DIRNAME,
+    EmnistDataLoaders,
+    EmnistDataset,
+)
+from .emnist_lines_dataset import (
+    construct_image_from_string,
+    EmnistLinesDataset,
+    get_samples_by_character,
+)
+from .sentence_generator import SentenceGenerator
+from .util import Transpose
 
-__all__ = ["EmnistDataLoader"]
+__all__ = [
+    "construct_image_from_string",
+    "DATA_DIRNAME",
+    "EmnistDataset",
+    "EmnistDataLoaders",
+    "EmnistLinesDataset",
+    "get_samples_by_character",
+    "SentenceGenerator",
+    "Transpose",
+]