Use tensorflow 2.15 (#312)

tensorflow · Apr 24, 2024 · 05aa080 · 05aa080
1 parent 8fcd5be
commit 05aa080
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -94,8 +94,8 @@ other *untested* combinations may also work.
 
 tensorflow-transform                                                            | apache-beam[gcp] | pyarrow | tensorflow        | tensorflow-metadata | tfx-bsl |
 ------------------------------------------------------------------------------- | -----------------| --------|-------------------|---------------------|---------|
-[GitHub master](https://github.com/tensorflow/transform/blob/master/RELEASE.md) | 2.47.0           | 10.0.0  | nightly (2.x)     | 1.15.0              | 1.15.0  |
-[1.15.0](https://github.com/tensorflow/transform/blob/v1.15.0/RELEASE.md)       | 2.47.0           | 10.0.0  | 2.14              | 1.15.0              | 1.15.0  |
+[GitHub master](https://github.com/tensorflow/transform/blob/master/RELEASE.md) | 2.47.0           | 10.0.0  | nightly (2.x)     | 1.15.0              | 1.15.1  |
+[1.15.0](https://github.com/tensorflow/transform/blob/v1.15.0/RELEASE.md)       | 2.47.0           | 10.0.0  | 2.15              | 1.15.0              | 1.15.1  |
 [1.14.0](https://github.com/tensorflow/transform/blob/v1.14.0/RELEASE.md)       | 2.47.0           | 10.0.0  | 2.13              | 1.14.0              | 1.14.0  |
 [1.13.0](https://github.com/tensorflow/transform/blob/v1.13.0/RELEASE.md)       | 2.41.0           | 6.0.0   | 2.12              | 1.13.1              | 1.13.0  |
 [1.12.0](https://github.com/tensorflow/transform/blob/v1.12.0/RELEASE.md)       | 2.41.0           | 6.0.0   | 2.11              | 1.12.0              | 1.12.0  |

diff --git a/RELEASE.md b/RELEASE.md
@@ -10,7 +10,7 @@
     (previously was 16.04).
 *   Explicitly use Keras 2 or `tf_keras`` if Keras 3 is installed.
 *   Added python 3.11 support.
-*   Depends on `tensorflow~=2.15`.
+*   Depends on `tensorflow 2.15`.
 *   Enable passing `tf.saved_model.SaveOptions` to model saving functionality.
 *   Census and sentiment examples updated to only use Keras instead of
     estimator.

diff --git a/docs/install.md b/docs/install.md
@@ -93,8 +93,8 @@ other *untested* combinations may also work.
 
 tensorflow-transform                                                            | apache-beam[gcp] | pyarrow | tensorflow        | tensorflow-metadata | tfx-bsl
 ------------------------------------------------------------------------------- | ---------------- | ------- | ----------------- | ------------------- | -------
-[GitHub master](https://github.com/tensorflow/transform/blob/master/RELEASE.md) | 2.47.0           | 10.0.0  | nightly (2.x)     | 1.15.0              | 1.15.0
-[1.15.0](https://github.com/tensorflow/transform/blob/v1.15.0/RELEASE.md)       | 2.47.0           | 10.0.0  | 2.14              | 1.15.0              | 1.15.0
+[GitHub master](https://github.com/tensorflow/transform/blob/master/RELEASE.md) | 2.47.0           | 10.0.0  | nightly (2.x)     | 1.15.0              | 1.15.1
+[1.15.0](https://github.com/tensorflow/transform/blob/v1.15.0/RELEASE.md)       | 2.47.0           | 10.0.0  | 2.15              | 1.15.0              | 1.15.1
 [1.14.0](https://github.com/tensorflow/transform/blob/v1.14.0/RELEASE.md)       | 2.47.0           | 10.0.0  | 2.13              | 1.14.0              | 1.14.0
 [1.13.0](https://github.com/tensorflow/transform/blob/v1.13.0/RELEASE.md)       | 2.41.0           | 6.0.0   | 2.12              | 1.13.1              | 1.13.0
 [1.12.0](https://github.com/tensorflow/transform/blob/v1.12.0/RELEASE.md)       | 2.41.0           | 6.0.0   | 2.11              | 1.12.0              | 1.12.0

diff --git a/setup.py b/setup.py
@@ -50,13 +50,13 @@ def _make_required_install_packages():
       'protobuf>=3.20.3,<5;python_version<"3.11"',
       'pyarrow>=10,<11',
       'pydot>=1.2,<2',
-      'tensorflow~=2.15',
+      'tensorflow>=2.15,<2.16',
       'tensorflow-metadata' + select_constraint(
           default='>=1.15.0,<1.16.0',
           nightly='>=1.16.0.dev',
           git_master='@git+https://github.com/tensorflow/metadata@master'),
       'tfx-bsl' + select_constraint(
-          default='>=1.15.0,<1.16.0',
+          default='>=1.15.1,<1.16.0',
           nightly='>=1.16.0.dev',
           git_master='@git+https://github.com/tensorflow/tfx-bsl@master'),
   ]