source/src/speculators/cli/train.py

"""Train command — thin typer wrapper around TrainConfig argument resolution."""

import typer


def train_command(ctx: typer.Context) -> None:
    """Train a speculator model.

    All arguments are forwarded to the training config resolver
    (use ``--help`` to see every available flag). Pass ``--config``
    with a YAML file, or use ``--key=value`` flags directly.

    \b
    Examples:
        speculators train --config config.yaml
        speculators train --verifier-name-or-path meta-llama/... \\
            --data-path ./output --speculator-type eagle3
        torchrun --nproc_per_node=4 -m speculators.train \\
            --config config.yaml
    """
    from speculators.train.cli import main  # noqa: PLC0415
    from speculators.train.config import TrainConfig  # noqa: PLC0415

    main(TrainConfig.resolve(ctx.args))