Depend on protobuf 3.20.3 for Python 3.9 and 3.10 and on 4.25.2 for Python 3.11.

PiperOrigin-RevId: 611585502

Author: iindyk

RELEASE.md

@@ -16,8 +16,10 @@
 *   Enable passing `tf.saved_model.SaveOptions` to model saving functionality.
 *   Census and sentiment examples updated to only use Keras instead of
     estimator.
-*   Depends on `apache-beam>=2.53,<3`.
-*   Depends on `protobuf>=4.25.2,<5`.
+*   Depends on `apache-beam[gcp]>=2.53.0,<3` for Python 3.11 and on 
+    `apache-beam[gcp]>=2.47.0,<3` for 3.9 and 3.10.
+*   Depends on `protobuf>=4.25.2,<5` for Python 3.11 and on `protobuf>3.20.3,<5`
+    for 3.9 and 3.10.
 
 ## Breaking Changes
 

setup.py

@@ -13,11 +13,15 @@
 # limitations under the License.
 """Package Setup script for tf.Transform."""
 import os
+import sys
 
 from setuptools import find_packages
 from setuptools import setup
 
 
+_IS_PY311 = sys.version_info >= (3, 11)
+
+
 def select_constraint(default, nightly=None, git_master=None):
   """Select dependency constraint based on TFX_DEPENDENCY_SELECTOR env var."""
   selector = os.environ.get('TFX_DEPENDENCY_SELECTOR')
@@ -43,9 +47,9 @@ 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.53,<3',
+      f'apache-beam[gcp]>={"2.53" if _IS_PY311 else "2.47"},<3',
       'numpy>=1.22.0',
-      'protobuf>=4.25.2,<5',
+      f'protobuf>={"4.25.2" if _IS_PY311 else "3.20.3"},<5',
       'pyarrow>=10,<11',
       'pydot>=1.2,<2',
       'tensorflow>=2.15,<3',