Skip to content

Commit 64b1f2a

Browse files
authored
[tests/requirements_tensorflow] Fix jax version for python 3.9 & NumPy conflict on Py 3.12 on MacOS (openvinotoolkit#27917)
### Details: Fix one of the blockers for dropping openvino-dev in internal CI --------- Signed-off-by: Alina Kladieva <alina.kladieva@intel.com>
1 parent c975788 commit 64b1f2a

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

tests/requirements_tensorflow

+6-4
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
11
# test ovc with NumPy 2.x on Ubuntu 24 with default Python 3.12
22
# test against NumPy 1.x with older Python versions
33
# tensorflow-intel 2.18.0 depends on numpy<2.1.0 and >=1.26.0
4-
numpy==1.26.4; python_version < "3.12"
5-
numpy==2.0.2; python_version >= "3.12"
4+
# tensorflow 2.16.2 depends on numpy<2.0.0 and >=1.26.0; python_version >= "3.12"
5+
numpy==1.26.4; python_version < "3.12" or platform_system == "Darwin" and platform_machine == "x86_64"
6+
numpy==2.0.2; python_version >= "3.12" and (platform_system != "Darwin" or platform_machine != "x86_64")
67
pytest==7.0.1
78
pytest-xdist[psutil]==3.6.1
89
pytest-html==4.1.1
@@ -16,7 +17,8 @@ wrapt==1.15.0; python_version >= "3.12"
1617
# tensorflow-text is not available for both Windows and ARM platforms
1718
tensorflow-text==2.18.0; python_version < "3.12" and platform_system == "Linux" and platform_machine == "x86_64"
1819
tensorflow-hub==0.16.1
19-
jax==0.4.35; platform_system != "Darwin" or platform_machine != "x86_64"
20+
jax==0.4.35; (platform_system != "Darwin" or platform_machine != "x86_64") and python_version > "3.9"
2021
# tensorflow 2.16.2 depends on ml-dtypes~=0.3.1 and jax 0.4.35 depends on ml-dtypes>=0.4.0
21-
jax==0.4.33; platform_system == "Darwin" and platform_machine == "x86_64"
22+
jax==0.4.33; (platform_system == "Darwin" and platform_machine == "x86_64") and python_version > "3.9"
23+
jax==0.4.30; python_version <= "3.9"
2224
defusedxml==0.7.1

0 commit comments

Comments
 (0)