Skip to content

Commit 083dcb0

Browse files
committed
Updated pyproject.toml
Replaced default torch installation with cpu version only
1 parent 7710c30 commit 083dcb0

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

pyproject.toml

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ testing = [
6161
"tensorflow_text>=2.11.0; platform_system!='Darwin' and python_version < '3.13'",
6262
"tensorflow_datasets",
6363
"tensorflow>=2.12.0; python_version<'3.13'", # to fix Numpy np.bool8 deprecation error
64-
"tensorflow>=2.20.0; python_version>='3.13'",
64+
"tensorflow>=2.20.0; python_version>='3.13' and python_version<'3.14'",
6565
"torch",
6666
"treescope>=0.1.1; python_version>='3.10'",
6767
"cloudpickle>=3.0.0",
@@ -237,3 +237,11 @@ quote-style = "single"
237237
[tool.uv]
238238
# Ignore uv.lock and always upgrade the package to the latest
239239
upgrade-package = ["jax", "jaxlib", "orbax-checkpoint"]
240+
241+
[tool.uv.sources]
242+
torch = { index = "pytorch" }
243+
244+
[[tool.uv.index]]
245+
name = "pytorch"
246+
url = "https://download.pytorch.org/whl/cpu"
247+
explicit = true

0 commit comments

Comments
 (0)