code/job_bootstrap.py

#!/usr/bin/env python3
"""Install the pinned KVPress revision, then run the scaled RULER evaluator."""

from __future__ import annotations

import os
import subprocess
import sys


KVPRESS_ARCHIVE = (
    "https://github.com/NVIDIA/kvpress/archive/"
    "8bb3315aa552d2d0b33f38ef0835e68cfa49a11a.tar.gz"
)


def main() -> None:
    subprocess.check_call(
        [sys.executable, "-m", "pip", "install", "--no-cache-dir", KVPRESS_ARCHIVE]
    )
    script = "/mnt/code/ruler_scaled_eval.py"
    os.execv(sys.executable, [sys.executable, script, *sys.argv[1:]])


if __name__ == "__main__":
    main()