src/dataset/__init__.py

"""Atlas dataset module."""

from .atlas_dataset import AtlasDataset, atlas_collate_fn, make_atlas_collate_fn

__all__ = [
    "AtlasDataset",
    "atlas_collate_fn",
    "make_atlas_collate_fn",
]