From 6cf7ca531906a82772bf63ec9d0dd4b9c1ea7760 Mon Sep 17 00:00:00 2001
From: David Huggins-Daines <dhd@ecolingui.ca>
Date: Sun, 17 Nov 2024 08:23:32 -0500
Subject: [PATCH] fix: avoid downloading CUDA junk

---
 pyproject.toml | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index cde854f..d72d08d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -59,6 +59,16 @@ addopts = "--cov=alexi --cov-report html"
 [tool.hatch.envs.default]
 features = [ "dev", "api", "yolo" ]
 
+[tool.hatch.envs.default.env-vars]
+# Avoid downloading gigabytes of CUDA junk
+PIP_EXTRA_INDEX_URL = "https://download.pytorch.org/whl/cpu"
+UV_EXTRA_INDEX_URL = "https://download.pytorch.org/whl/cpu"
+
+[tool.hatch.envs.gpu.env-vars]
+# Disable PyTorch CPU respository
+PIP_EXTRA_INDEX_URL = ""
+UV_EXTRA_INDEX_URL = ""
+
 [tool.hatch.envs.default.scripts]
 test = [
   "pytest", "coverage html"