File tree Expand file tree Collapse file tree 1 file changed +9
-1
lines changed Expand file tree Collapse file tree 1 file changed +9
-1
lines changed Original file line number Diff line number Diff 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
239239upgrade-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
You can’t perform that action at this time.
0 commit comments