diff --git a/requirements.txt b/requirements.txt index 6c6afff6842..1e7b1c19438 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,12 +1,12 @@ # Tensorflow. -tensorflow-cpu~=2.18.0;sys_platform != 'darwin' -tensorflow~=2.18.0;sys_platform == 'darwin' +tensorflow-cpu~=2.19.0;sys_platform != 'darwin' tf_keras tf2onnx # Torch. --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.6.0+cpu +torch==2.6.0+cpu;sys_platform != 'darwin' +torch==2.6.0;sys_platform == 'darwin' torch-xla==2.6.0;sys_platform != 'darwin' # Jax. @@ -15,5 +15,8 @@ torch-xla==2.6.0;sys_platform != 'darwin' jax[cpu]==0.5.0 flax +# pre-commit checks (formatting, linting, etc.) +pre-commit + # Common deps. -r requirements-common.txt