code/self_healing_finetuning/training/hf_job_entrypoint.py

from __future__ import annotations

import os
import subprocess
import sys
from pathlib import Path


def main() -> int:
    root = Path(__file__).resolve().parents[1]
    sys.path.insert(0, str(root))
    from training.hf_runtime_deps import install_runtime_packages

    trainer = root / "training" / "hf_fingpt_train.py"
    if not trainer.exists():
        print(f"missing trainer script: {trainer}", file=sys.stderr)
        return 2

    install_runtime_packages(include_trl=True)

    env = os.environ.copy()
    env["PYTHONPATH"] = str(root)
    command = [sys.executable, str(trainer), *sys.argv[1:]]
    return subprocess.call(command, env=env)


if __name__ == "__main__":
    raise SystemExit(main())