Working bash scripts etc.

1 files changed, 4 insertions, 9 deletions

diff --git a/src/text_recognizer/datasets/__init__.py b/src/text_recognizer/datasets/__init__.py
index 1b4cc59..05f74f6 100644
--- a/src/text_recognizer/datasets/__init__.py
+++ b/src/text_recognizer/datasets/__init__.py
@@ -1,29 +1,24 @@
 """Dataset modules."""
 from .emnist_dataset import (
-    _augment_emnist_mapping,
-    _load_emnist_essentials,
     DATA_DIRNAME,
-    EmnistDataLoaders,
     EmnistDataset,
+    EmnistMapper,
     ESSENTIALS_FILENAME,
 )
 from .emnist_lines_dataset import (
     construct_image_from_string,
-    EmnistLinesDataLoaders,
     EmnistLinesDataset,
     get_samples_by_character,
 )
-from .util import Transpose
+from .util import fetch_data_loaders, Transpose
 
 __all__ = [
-    "_augment_emnist_mapping",
-    "_load_emnist_essentials",
     "construct_image_from_string",
     "DATA_DIRNAME",
     "EmnistDataset",
-    "EmnistDataLoaders",
-    "EmnistLinesDataLoaders",
+    "EmnistMapper",
     "EmnistLinesDataset",
+    "fetch_data_loaders",
     "get_samples_by_character",
     "Transpose",
 ]