From b22bb021617178d564ad507da53de7eff5d00dc1 Mon Sep 17 00:00:00 2001 From: Roman Kazantsev Date: Tue, 28 Jan 2025 01:54:23 +0400 Subject: [PATCH] [TF FE][JAX FE] Test with JAX 0.5.0 and Keras 3.8.0 (#28701) **Details:** Test with latest JAX 0.5.0 and Keras 3.8.0 **Ticket:** TBD Signed-off-by: Kazantsev, Roman --- tests/requirements_jax | 2 +- tests/requirements_tensorflow | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/requirements_jax b/tests/requirements_jax index c392df4359bee3..ac939d34f84383 100644 --- a/tests/requirements_jax +++ b/tests/requirements_jax @@ -3,7 +3,7 @@ numpy==2.2.1; python_version >= "3.12" and (platform_system != "Darwin" or platf pytest==7.0.1 pytest-xdist[psutil]==3.6.1 pytest-html==4.1.1 -jax==0.4.38; (platform_system != "Darwin" or platform_machine != "x86_64") and python_version > "3.9" +jax==0.5.0; (platform_system != "Darwin" or platform_machine != "x86_64") and python_version > "3.9" # tensorflow 2.16.2 depends on ml-dtypes~=0.3.1 and jax 0.4.35 depends on ml-dtypes>=0.4.0 jax==0.4.33; (platform_system == "Darwin" and platform_machine == "x86_64") and python_version > "3.9" jax==0.4.30; python_version <= "3.9" diff --git a/tests/requirements_tensorflow b/tests/requirements_tensorflow index 5d699facad1c91..cd32808047d782 100644 --- a/tests/requirements_tensorflow +++ b/tests/requirements_tensorflow @@ -10,7 +10,7 @@ pytest-xdist[psutil]==3.6.1 pytest-html==4.1.1 transformers==4.45.1 # install exact keras version since tensorflow depends and has no upper bound for it -keras==3.6.0 +keras==3.8.0 tensorflow==2.18.0; platform_system != "Darwin" or platform_machine != "x86_64" tensorflow==2.16.2; platform_system == "Darwin" and platform_machine == "x86_64" # install explicit version of wrapt to avoid "this __dict__ descriptor does not support '_DictWrapper' objects" error from TensorFlow 2.18 @@ -18,7 +18,7 @@ wrapt==1.15.0; python_version >= "3.12" # tensorflow-text is not available for both Windows and ARM platforms tensorflow-text==2.18.0; python_version < "3.12" and platform_system == "Linux" and platform_machine == "x86_64" tensorflow-hub==0.16.1 -jax==0.4.38; (platform_system != "Darwin" or platform_machine != "x86_64") and python_version > "3.9" +jax==0.5.0; (platform_system != "Darwin" or platform_machine != "x86_64") and python_version > "3.9" # tensorflow 2.16.2 depends on ml-dtypes~=0.3.1 and jax 0.4.35 depends on ml-dtypes>=0.4.0 jax==0.4.33; (platform_system == "Darwin" and platform_machine == "x86_64") and python_version > "3.9" jax==0.4.30; python_version <= "3.9"