Merge pull request apache#31386 from tvalentyn/py312_tf

Don't install TF on Python 3.12, since there is no compatible version atm.
robertwb · May 24, 2024 · 1ed29e1 · 1ed29e1
2 parents dc6b72c + 697c18f
commit 1ed29e1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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"',