mogen/datasets/pipelines/__init__.py

from .compose import Compose
from .formatting import (
    to_tensor,
    ToTensor,
    Transpose,
    Collect,
    WrapFieldsToLists
)
from .transforms import (
    Crop,
    RandomCrop,
    Normalize
)

__all__ = [
    'Compose', 'to_tensor', 'Transpose', 'Collect', 'WrapFieldsToLists', 'ToTensor',
    'Crop', 'RandomCrop', 'Normalize'
]