depthdif_dataset/__init__.py

"""Standalone PyTorch loaders for the DepthDif Hugging Face dataset."""

from .dataloaders import (
    DepthTileDataModule,
    build_dataloader,
    build_train_val_dataloaders,
    split_dataset,
)
from .dataset import (
    ArgoGeoTIFFGriddedPatchDataset,
    ArgoGeoTIFFProfileStore,
    GeoTIFFPatchIndex,
    GeoTIFFRasterStore,
)
from .normalizations import salinity_normalize, temperature_normalize

__all__ = [
    "ArgoGeoTIFFGriddedPatchDataset",
    "ArgoGeoTIFFProfileStore",
    "DepthTileDataModule",
    "GeoTIFFPatchIndex",
    "GeoTIFFRasterStore",
    "build_dataloader",
    "build_train_val_dataloaders",
    "salinity_normalize",
    "split_dataset",
    "temperature_normalize",
]