From 64b1f2a7632089544575f076a394580824e58338 Mon Sep 17 00:00:00 2001 From: Alina Kladieva Date: Wed, 4 Dec 2024 23:51:44 +0100 Subject: [PATCH] [tests/requirements_tensorflow] Fix jax version for python 3.9 & NumPy conflict on Py 3.12 on MacOS (#27917) ### Details: Fix one of the blockers for dropping openvino-dev in internal CI --------- Signed-off-by: Alina Kladieva --- tests/requirements_tensorflow | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/tests/requirements_tensorflow b/tests/requirements_tensorflow index 954bba7944245f..5369b0135f7618 100644 --- a/tests/requirements_tensorflow +++ b/tests/requirements_tensorflow @@ -1,8 +1,9 @@ # test ovc with NumPy 2.x on Ubuntu 24 with default Python 3.12 # test against NumPy 1.x with older Python versions # tensorflow-intel 2.18.0 depends on numpy<2.1.0 and >=1.26.0 -numpy==1.26.4; python_version < "3.12" -numpy==2.0.2; python_version >= "3.12" +# tensorflow 2.16.2 depends on numpy<2.0.0 and >=1.26.0; python_version >= "3.12" +numpy==1.26.4; python_version < "3.12" or platform_system == "Darwin" and platform_machine == "x86_64" +numpy==2.0.2; python_version >= "3.12" and (platform_system != "Darwin" or platform_machine != "x86_64") pytest==7.0.1 pytest-xdist[psutil]==3.6.1 pytest-html==4.1.1 @@ -16,7 +17,8 @@ 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.35; platform_system != "Darwin" or platform_machine != "x86_64" +jax==0.4.35; (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" +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" defusedxml==0.7.1