src/datasets/understanding/__init__.py

from .llava_datasets import LLaVADataset
from .caption_datasets import CaptionDataset

__all__ = ['LLaVADataset', 'CaptionDataset']