Bump up minimum apache-beam version to 2.40.0 to support new run_inference API.

PiperOrigin-RevId: 464616132

Author: tf-transform-team

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   |

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
 

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

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',