dataset/__init__.py

from .baselines import BASELINES, canonical, by_baseline
from .datasets import (
    ORDERINGS,
    Sample,
    Task,
    load_task,
    load_vsr,
    load_youtube,
    missing_images,
    normalize_choice_letter,
    normalize_letter,
    normalize_ordering,
    normalize_true_false,
)

__all__ = [
    "BASELINES",
    "ORDERINGS",
    "Sample",
    "Task",
    "by_baseline",
    "canonical",
    "load_task",
    "load_vsr",
    "load_youtube",
    "missing_images",
    "normalize_choice_letter",
    "normalize_letter",
    "normalize_ordering",
    "normalize_true_false",
]