[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 <[email protected]>
openvinotoolkit · Jan 27, 2025 · b22bb02 · b22bb02
1 parent 7a9d23e
commit b22bb02
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -10,15 +10,15 @@ 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
 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"