1 files changed, 7 insertions, 0 deletions
diff --git a/src/text_recognizer/datasets/__init__.py b/src/text_recognizer/datasets/__init__.py
index a8c46c4..1b4cc59 100644
--- a/src/text_recognizer/datasets/__init__.py
+++ b/src/text_recognizer/datasets/__init__.py
@@ -1,21 +1,28 @@
 """Dataset modules."""
 from .emnist_dataset import (
+    _augment_emnist_mapping,
+    _load_emnist_essentials,
     DATA_DIRNAME,
     EmnistDataLoaders,
     EmnistDataset,
+    ESSENTIALS_FILENAME,
 )
 from .emnist_lines_dataset import (
     construct_image_from_string,
+    EmnistLinesDataLoaders,
     EmnistLinesDataset,
     get_samples_by_character,
 )
 from .util import Transpose
 
 __all__ = [
+    "_augment_emnist_mapping",
+    "_load_emnist_essentials",
     "construct_image_from_string",
     "DATA_DIRNAME",
     "EmnistDataset",
     "EmnistDataLoaders",
+    "EmnistLinesDataLoaders",
     "EmnistLinesDataset",
     "get_samples_by_character",
     "Transpose",