fix: avoid downloading CUDA junk

dhdaines · Nov 17, 2024 · 6cf7ca5 · 6cf7ca5
1 parent 4b2b10b
commit 6cf7ca5
Showing 1 changed file with 10 additions and 0 deletions.
diff --git 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"