Finished updating data loading functions, bumped version for release

Author: gperdrizet

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "image-classification-tools"
-version = "0.5.5"
+version = "0.5.6"
 description = "A lightweight PyTorch toolkit for building and training image classification models"
 authors = ["gperdrizet <[email protected]>"]
 readme = "src/image_classification_tools/README.md"

src/image_classification_tools/pytorch/__init__.py

@@ -1,7 +1,7 @@
 '''PyTorch utilities for image classification.'''
 
 from image_classification_tools.pytorch.data import (
-    load_datasets,
+    load_dataset,
     prepare_splits,
     create_dataloaders,
     generate_augmented_data
@@ -24,7 +24,7 @@
 
 __all__ = [
     # Data loading and preprocessing
-    'load_datasets',
+    'load_dataset',
     'prepare_splits',
     'create_dataloaders',
     'generate_augmented_data',