ContraCLIP/models/genforce/datasets/__init__.py

# python3.7
"""Collects datasets and data loaders."""

from .datasets import BaseDataset
from .dataloaders import IterDataLoader

__all__ = ['BaseDataset', 'IterDataLoader']