From 6876734da12bf7ab2a7e6c09947d87a0c431914e Mon Sep 17 00:00:00 2001 From: Valentyn Tymofieiev Date: Thu, 23 May 2024 17:01:40 -0700 Subject: [PATCH 1/2] Don't install TF on Python 3.12. --- sdks/python/setup.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/sdks/python/setup.py b/sdks/python/setup.py index 4a7b57db917b..728bdd9b8d39 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.11"', 'tensorflow-hub', # https://github.com/tensorflow/transform/issues/313 'tensorflow-transform;python_version<"3.11"', From 697c18fd6a5b44319b858915d560ca6558f1dd88 Mon Sep 17 00:00:00 2001 From: tvalentyn Date: Fri, 24 May 2024 08:50:52 -0700 Subject: [PATCH 2/2] Update setup.py --- sdks/python/setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdks/python/setup.py b/sdks/python/setup.py index 728bdd9b8d39..3a8856583153 100644 --- a/sdks/python/setup.py +++ b/sdks/python/setup.py @@ -490,7 +490,7 @@ def get_portability_package_data(): 'skl2onnx', # 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.11"', + 'tensorflow<2.16.0;python_version<"3.12"', 'tensorflow-hub', # https://github.com/tensorflow/transform/issues/313 'tensorflow-transform;python_version<"3.11"',