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

PiperOrigin-RevId: 612880486

Author: tfx-copybara

RELEASE.md

@@ -10,6 +10,8 @@
     (previously was 16.04).
 *   Depends on `tensorflow>=2.15.0,<3`.
 *   Bumped the minimum bazel version required to build `struct2tensor` to 6.1.0.
+*   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

@@ -15,12 +15,15 @@
 """Setup for pip package."""
 
 import os
+import sys
 
 from setuptools import find_packages
 from setuptools import setup
 from setuptools.command.install import install
 from setuptools.dist import Distribution
 
+_IS_PY311 = sys.version_info >= (3, 11)
+
 
 # struct2tensor is not a purelib. However because of the extension module is not
 # built by setuptools, it will be incorrectly treated as a purelib. The
@@ -77,7 +80,7 @@ def select_constraint(default, nightly=None, git_master=None):
         # TODO(b/263060885): Remove the explicit numpy dependency once TF works
         # with numpy>=1.24.
         'numpy>=1.22',
-        'protobuf>=3.20.3,<5',
+        f'protobuf>={"4.25.2" if _IS_PY311 else "3.20.3"},<5',
         'tensorflow>=2.15,<3',
         'tensorflow-metadata' + select_constraint(
             default='>=1.14.0,<1.15.0',