source/src/strata/experiments/__init__.py

"""Reproducible multi-stage training experiments (curricula) for STRATA."""

from strata.experiments.curriculum import (
    Curriculum,
    ResolvedStage,
    Stage,
    build_stage_command,
    latest_checkpoint,
    load_curriculum,
    plan_actions,
    resolve_stages,
)
from strata.experiments.model_registry import (
    ModelRegistry,
    ModelRegistryEntry,
    load_model_registry,
    validate_model_registry,
)

__all__ = [
    "Curriculum",
    "ModelRegistry",
    "ModelRegistryEntry",
    "ResolvedStage",
    "Stage",
    "build_stage_command",
    "latest_checkpoint",
    "load_model_registry",
    "load_curriculum",
    "plan_actions",
    "resolve_stages",
    "validate_model_registry",
]