training/cli.py

"""Unified CLI entry point for SAE training."""

from __future__ import annotations

import sys
from pathlib import Path
from typing import Optional

ROOT = Path(__file__).resolve().parents[1]
if str(ROOT) not in sys.path:
    sys.path.insert(0, str(ROOT))

from log_config import setup_logging
from training.config import parse_args
from training.trainer import Trainer


def main(default_iqa_model: Optional[str] = None) -> None:
    setup_logging()
    args, backend = parse_args(default_iqa_model=default_iqa_model)
    Trainer(backend, args).run()


if __name__ == "__main__":
    main()