"""Create the pinned inference environment and build its CUDA convolution kernel."""

import argparse
import os
import subprocess
import sys
from pathlib import Path

from cuda_toolchain import configure


def main():
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--venv", default=".venv", help="Destination environment")
    args = parser.parse_args()
    root = Path(__file__).resolve().parent
    venv = Path(args.venv).resolve()
    configure(install=True)
    subprocess.run([sys.executable, "-m", "pip", "install", "uv==0.12.17"], check=True)
    subprocess.run(
        [sys.executable, "-m", "uv", "sync", "--project", str(root), "--locked",
         "--python", "3.13.15"],
        env=os.environ | {"UV_PROJECT_ENVIRONMENT": str(venv)}, check=True,
    )
    subprocess.run(
        [str(venv / "bin/python"), str(root / "build_causal_conv.py"),
         "--output", str(venv / "kernel-build"), "--install"], check=True,
    )
    print("Ready. Run examples with " + str(venv / "bin/python"))


if __name__ == "__main__":
    main()