Skip to content

Commit 757272e

Browse files
Use CPU-only PyTorch index to speed up dev installation (#1556)
1 parent 1c13133 commit 757272e

1 file changed

Lines changed: 17 additions & 0 deletions

File tree

pyproject.toml

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,11 @@ optional-dependencies.dev = [
6868
"sphinx-substitution-extensions==2026.1.12",
6969
"sphinxcontrib-spelling==8.0.2",
7070
"sybil==9.3.0",
71+
# Listed explicitly (despite being transitive via vws-python-mock) so that
72+
# [tool.uv.sources] can redirect to the CPU-only PyTorch index.
73+
# See: https://vws-python.github.io/vws-python-mock/installation.html#faster-installation
74+
"torch>=2.5.1",
75+
"torchvision>=0.20.1",
7176
"ty==0.0.17",
7277
"types-requests==2.32.4.20260107",
7378
"vulture==2.14",
@@ -99,6 +104,11 @@ bdist_wheel.universal = true
99104
# Code to match this is in ``conf.py``.
100105
version_scheme = "post-release"
101106

107+
[tool.uv]
108+
sources.torch = { index = "pytorch-cpu" }
109+
sources.torchvision = { index = "pytorch-cpu" }
110+
index = [ { name = "pytorch-cpu", url = "https://download.pytorch.org/whl/cpu", explicit = true } ]
111+
102112
[tool.ruff]
103113
line-length = 79
104114
lint.select = [
@@ -270,6 +280,13 @@ pep621_dev_dependency_groups = [
270280
"dev",
271281
"release",
272282
]
283+
# torch and torchvision are listed explicitly in dev deps to allow
284+
# [tool.uv.sources] to redirect them to the CPU-only PyTorch index,
285+
# but they are not directly imported in the vws-auth-tools source code.
286+
per_rule_ignores.DEP002 = [
287+
"torch",
288+
"torchvision",
289+
]
273290

274291
[tool.pyproject-fmt]
275292
indent = 4

0 commit comments

Comments
 (0)