dataforge/datasets/__init__.py

"""Dataset loading helpers for DataForge."""

from dataforge.datasets.real_world import (
    DatasetDownloadError,
    GroundTruthCell,
    RealWorldDataset,
    load_real_world_dataset,
)
from dataforge.datasets.registry import (
    DATASET_REGISTRY,
    DatasetMetadata,
    HeaderMismatch,
    get_dataset_metadata,
)

__all__ = [
    "DATASET_REGISTRY",
    "DatasetDownloadError",
    "DatasetMetadata",
    "GroundTruthCell",
    "HeaderMismatch",
    "RealWorldDataset",
    "get_dataset_metadata",
    "load_real_world_dataset",
]