Skip to content

Commit 92fb88a

Browse files
committed
Fix PyTorch and vispy dependency resolution for all platforms
- Use pytorch-cpu index for macOS, pytorch-cu124 for Linux/Windows - Remove platform restriction from vispy override (now applies to all platforms) - Regenerate lock file with correct multi-platform resolution
1 parent 1151be7 commit 92fb88a

2 files changed

Lines changed: 338 additions & 57 deletions

File tree

pyproject.toml

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -31,28 +31,33 @@ environments = [
3131
"sys_platform == 'linux'",
3232
]
3333
override-dependencies = [
34-
"torch>=2.6.0; sys_platform == 'darwin'",
35-
"torchvision>=0.21.0; sys_platform == 'darwin'",
36-
"torchaudio>=2.6.0; sys_platform == 'darwin'",
37-
"vispy>=0.15.2; sys_platform == 'darwin'",
34+
"vispy>=0.15.2",
3835
]
3936

4037
[tool.uv.sources]
4138
biosignal-device-interface = { git = "https://github.com/NsquaredLab/Biosignal-Device-Interface.git" }
4239
sphinxcontrib-pdfembed = { git = "https://github.com/SuperKogito/sphinxcontrib-pdfembed.git" }
4340
myoverse = { git = "https://github.com/NsquaredLab/MyoVerse.git" }
4441
torch = [
45-
{ index = "torch-gpu", marker = "sys_platform == 'win32'" },
42+
{ index = "pytorch-cpu", marker = "sys_platform == 'darwin'" },
43+
{ index = "pytorch-cu124", marker = "sys_platform != 'darwin'" },
4644
]
4745
torchvision = [
48-
{ index = "torch-gpu", marker = "sys_platform == 'win32'" },
46+
{ index = "pytorch-cpu", marker = "sys_platform == 'darwin'" },
47+
{ index = "pytorch-cu124", marker = "sys_platform != 'darwin'" },
4948
]
5049
torchaudio = [
51-
{ index = "torch-gpu", marker = "sys_platform == 'win32'" },
50+
{ index = "pytorch-cpu", marker = "sys_platform == 'darwin'" },
51+
{ index = "pytorch-cu124", marker = "sys_platform != 'darwin'" },
5252
]
5353

5454
[[tool.uv.index]]
55-
name = "torch-gpu"
55+
name = "pytorch-cpu"
56+
url = "https://download.pytorch.org/whl/cpu"
57+
explicit = true
58+
59+
[[tool.uv.index]]
60+
name = "pytorch-cu124"
5661
url = "https://download.pytorch.org/whl/cu124"
5762
explicit = true
5863

0 commit comments

Comments
 (0)