File tree Expand file tree Collapse file tree 2 files changed +608
-101
lines changed
Expand file tree Collapse file tree 2 files changed +608
-101
lines changed Original file line number Diff line number Diff line change @@ -10,16 +10,28 @@ dependencies = [
1010 " numpy>=1.24.0" ,
1111 " pytest>=7.0.0" ,
1212 " pytest-html>=4.0.0" ,
13+ " matplotlib>=3.5.0" ,
1314]
1415
16+ [[tool .uv .index ]]
17+ name = " pytorch-cpu"
18+ url = " https://download.pytorch.org/whl/cpu"
19+ explicit = true
20+
1521[[tool .uv .index ]]
1622name = " pytorch-cu118"
1723url = " https://download.pytorch.org/whl/cu118"
1824explicit = true
1925
2026[tool .uv .sources ]
21- torch = [{ index = " pytorch-cu118" }]
22- torchvision = [{ index = " pytorch-cu118" }]
27+ torch = [
28+ { index = " pytorch-cpu" , marker = " sys_platform == 'darwin'" },
29+ { index = " pytorch-cu118" , marker = " sys_platform == 'linux'" }
30+ ]
31+ torchvision = [
32+ { index = " pytorch-cpu" , marker = " sys_platform == 'darwin'" },
33+ { index = " pytorch-cu118" , marker = " sys_platform == 'linux'" }
34+ ]
2335
2436[tool .hatch .build .targets .wheel ]
2537packages = []
You can’t perform that action at this time.
0 commit comments