Merge pull request #57 from Xiao-Chenguang/torch

Config torch for all platform except macos
Xiao-Chenguang · Nov 21, 2024 · 962d7d9 · 962d7d9
2 parents 22da169 + 1013d4c
commit 962d7d9
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -48,5 +48,10 @@ dev-dependencies = [
     "torchvision>=0.19.1",
 ]
 
-[tool.ruff.lint.isort]
-known-third-party = ["wandb"]
+[tool.uv.sources]
+torch = [{ index = "pytorch-cu121", marker = "platform_system != 'Darwin'" }]
+
+[[tool.uv.index]]
+name = "pytorch-cu121"
+url = "https://download.pytorch.org/whl/cu121"
+explicit = true