diff --git a/sdks/python/setup.py b/sdks/python/setup.py
index 4a7b57db917b..3a8856583153 100644
--- a/sdks/python/setup.py
+++ b/sdks/python/setup.py
@@ -488,8 +488,9 @@ def get_portability_package_data():
               'onnxruntime',
               'sentence-transformers',
               'skl2onnx',
-              # https://github.com/apache/beam/issues/31294
-              'tensorflow<2.16.0',
+              # Support TF 2.16.0: https://github.com/apache/beam/issues/31294
+              # Once TF version is unpinned, also don't restrict Python version.
+              'tensorflow<2.16.0;python_version<"3.12"',
               'tensorflow-hub',
               # https://github.com/tensorflow/transform/issues/313
               'tensorflow-transform;python_version<"3.11"',