Bump up minimum apache-beam version to 2.40.0 to support new run_infe…

…rence API. PiperOrigin-RevId: 464616132
tensorflow · Aug 1, 2022 · d627663 · d627663
1 parent 19c317f
commit d627663
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -96,7 +96,7 @@ 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.38.0           | 5.0.0   | nightly (1.x/2.x) | 1.9.0               | 1.9.0   |
+[GitHub master](https://github.com/tensorflow/transform/blob/master/RELEASE.md) | 2.40.0           | 5.0.0   | nightly (1.x/2.x) | 1.9.0               | 1.9.0   |
 [1.9.0](https://github.com/tensorflow/transform/blob/v1.9.0/RELEASE.md)         | 2.38.0           | 5.0.0   | 1.15.5 / 2.9      | 1.9.0               | 1.9.0   |
 [1.8.0](https://github.com/tensorflow/transform/blob/v1.8.0/RELEASE.md)         | 2.38.0           | 5.0.0   | 1.15.5 / 2.8      | 1.8.0               | 1.8.0   |
 [1.7.0](https://github.com/tensorflow/transform/blob/v1.7.0/RELEASE.md)         | 2.36.0           | 5.0.0   | 1.15.5 / 2.8      | 1.7.0               | 1.7.0   |

diff --git a/RELEASE.md b/RELEASE.md
@@ -10,6 +10,7 @@
     cloned PTransforms in deep copy optimization. The reason of adding these
     resource hints is to prevent root Reads that are generated from deep copy
     being merged due to common subexpression elimination.
+*   Depends on `apache-beam[gcp]>=2.40,<3`.
 
 ## Breaking Changes
 

diff --git a/docs/install.md b/docs/install.md
@@ -95,7 +95,7 @@ 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.38.0           | 5.0.0   | nightly (1.x/2.x) | 1.9.0               | 1.9.0
+[GitHub master](https://github.com/tensorflow/transform/blob/master/RELEASE.md) | 2.40.0           | 5.0.0   | nightly (1.x/2.x) | 1.9.0               | 1.9.0
 [1.9.0](https://github.com/tensorflow/transform/blob/v1.9.0/RELEASE.md)         | 2.38.0           | 5.0.0   | 1.15.5 / 2.9      | 1.9.0               | 1.9.0
 [1.8.0](https://github.com/tensorflow/transform/blob/v1.8.0/RELEASE.md)         | 2.38.0           | 5.0.0   | 1.15.5 / 2.8      | 1.8.0               | 1.8.0
 [1.7.0](https://github.com/tensorflow/transform/blob/v1.7.0/RELEASE.md)         | 2.36.0           | 5.0.0   | 1.15.5 / 2.8      | 1.7.0               | 1.7.0

diff --git a/setup.py b/setup.py
@@ -43,7 +43,7 @@ def _make_required_install_packages():
   # protobuf) with TF and pyarrow version with tfx-bsl.
   return [
       'absl-py>=0.9,<2.0.0',
-      'apache-beam[gcp]>=2.38,<3',
+      'apache-beam[gcp]>=2.40,<3',
       'numpy>=1.16,<2',
       'protobuf>=3.13,<4',
       'pyarrow>=1,<6',