2024-04-02 20:19:37 +08:00
|
|
|
--- a/tensorflow/tools/pip_package/setup.py
|
|
|
|
|
+++ b/tensorflow/tools/pip_package/setup.py
|
2023-09-13 14:32:35 +08:00
|
|
|
@@ -86,13 +86,13 @@ REQUIRED_PACKAGES = [
|
2022-11-30 14:13:00 +08:00
|
|
|
'astunparse >= 1.6.0',
|
|
|
|
|
'flatbuffers >= 2.0',
|
|
|
|
|
# TODO(b/213222745) gast versions above 0.4.0 break TF's tests
|
|
|
|
|
- 'gast >= 0.2.1, <= 0.4.0',
|
|
|
|
|
+ 'gast >= 0.2.1',
|
2022-11-14 23:59:53 +08:00
|
|
|
'google_pasta >= 0.1.1',
|
|
|
|
|
'h5py >= 2.9.0',
|
2023-09-13 14:32:35 +08:00
|
|
|
# TODO(b/239052279): replace with external dependency on JAX repo once JAX
|
|
|
|
|
# no longer relies on TF.
|
|
|
|
|
- 'jax >= 0.3.15',
|
2022-11-14 23:59:53 +08:00
|
|
|
- 'libclang >= 13.0.0',
|
2023-09-13 14:32:35 +08:00
|
|
|
+ #'jax >= 0.3.15',
|
2022-11-14 23:59:53 +08:00
|
|
|
+ #'libclang >= 13.0.0',
|
2023-09-13 14:32:35 +08:00
|
|
|
# TODO(b/263178356): numpy 1.24 breaks TF's tests
|
|
|
|
|
'numpy >= 1.22, <= 1.24.3',
|
2022-11-14 23:59:53 +08:00
|
|
|
'opt_einsum >= 2.3.2',
|
2024-04-02 20:19:37 +08:00
|
|
|
@@ -108,12 +108,12 @@ REQUIRED_PACKAGES = [
|
|
|
|
|
'setuptools',
|
|
|
|
|
'six >= 1.12.0',
|
|
|
|
|
'termcolor >= 1.1.0',
|
|
|
|
|
- 'typing_extensions>=3.6.6,<4.6.0',
|
|
|
|
|
+ 'typing_extensions>=3.6.6,<=4.10.0',
|
2023-09-13 14:32:35 +08:00
|
|
|
# TODO(b/266362323): wrapt==1.15.0rc0 incompatible with TF 2.12.0 RC0 (and
|
|
|
|
|
# nightly, but works with TF 2.11)
|
|
|
|
|
- 'wrapt >= 1.11.0, <1.15',
|
|
|
|
|
- 'tensorflow-io-gcs-filesystem >= 0.23.1;platform_machine!="arm64" or ' +
|
|
|
|
|
- 'platform_system!="Darwin"',
|
|
|
|
|
+ 'wrapt >= 1.11.0',
|
|
|
|
|
+ #'tensorflow-io-gcs-filesystem >= 0.23.1;platform_machine!="arm64" or ' +
|
|
|
|
|
+ #'platform_system!="Darwin"',
|
2022-11-14 23:59:53 +08:00
|
|
|
# grpcio does not build correctly on big-endian machines due to lack of
|
|
|
|
|
# BoringSSL support.
|
|
|
|
|
# See https://github.com/tensorflow/tensorflow/issues/17882.
|
