From 1013d4ced5dba2ccbddf4bcb93454074c09d2ed9 Mon Sep 17 00:00:00 2001
From: Xiao-Chenguang <973990275@qq.com>
Date: Thu, 21 Nov 2024 16:34:57 +0000
Subject: [PATCH] config torch for all platform except macos

---
 pyproject.toml | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 71a288c..89a3924 100644
--- 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