swap

courtois-neuromod · Oct 22, 2024 · ff4fedd · ff4fedd
1 parent 7548242
commit ff4fedd
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -88,14 +88,13 @@ dependencies = [
     "pytest-cov==5.0.0", # Test coverage
 ]
 
-[[tool.uv.index]]
-name = "torch"
-url = "https://download.pytorch.org/whl/cu124"
-
 [tool.uv.sources]
-torch = { index = "torch" }
-torchaudio = { index = "torch" }
-torchvision = { index = "torch" }
+torch = { index = "torch-cu124" }
+torchaudio = { index = "torch-cu124" }
+torchvision = { index = "torch-cu124" }
+
+[tool.uv.indexes]
+torch-cu124 = "https://download.pytorch.org/whl/cu124"
 
 [tool.black]
 line-length = 79