opsd_utils/__init__.py

from opsd_utils.constants import MODE_GRPO, MODE_OPSD, MODE_SFT
from opsd_utils.mode_router import route_prompt_modes, expand_modes_to_completions
from opsd_utils.opsd_loss import compute_vlm_opsd_loss
from opsd_utils.privileged import build_privileged_context
from opsd_utils.recoverability import estimate_recoverable_flags
from opsd_utils.prompt_builder import build_teacher_prompt_batch

__all__ = [
    "MODE_GRPO",
    "MODE_OPSD",
    "MODE_SFT",
    "route_prompt_modes",
    "expand_modes_to_completions",
    "compute_vlm_opsd_loss",
    "build_privileged_context",
    "estimate_recoverable_flags",
    "build_teacher_prompt_batch",
]