Switch to PEP 508 style conditional dependencies.

PiperOrigin-RevId: 612943796

Author: iindyk

setup.py

@@ -13,15 +13,11 @@
 # 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')
@@ -47,9 +43,11 @@ def _make_required_install_packages():
   # protobuf) with TF and pyarrow version with tfx-bsl.
   return [
       'absl-py>=0.9,<2.0.0',
-      f'apache-beam[gcp]>={"2.53" if _IS_PY311 else "2.47"},<3',
+      'apache-beam[gcp]>=2.53,<3;python_version>="3.11"',
+      'apache-beam[gcp]>=2.47,<3;python_version<"3.11"',
       'numpy>=1.22.0',
-      f'protobuf>={"4.25.2" if _IS_PY311 else "3.20.3"},<5',
+      'protobuf>=4.25.2,<5;python_version>="3.11"',
+      'protobuf>=3.20.3,<5;python_version<"3.11"',
       'pyarrow>=10,<11',
       'pydot>=1.2,<2',
       'tensorflow>=2.15,<3',