data_proc/__init__.py

# __init__.py
__all__ = ["cross_entropy_dataset", "triplet_loss_dataset"]

from .cross_entropy_dataset import *
from .triplet_loss_dataset import *