This is an automated email from the ASF dual-hosted git repository. yhu pushed a commit to branch set-absl-py in repository https://gitbox.apache.org/repos/asf/beam.git
commit 61147c1ebb01c2a2e79b3bc4cb0155b6d752522c Author: Yi Hu <[email protected]> AuthorDate: Thu Feb 26 14:20:28 2026 -0500 Fix ml_base dependency --- sdks/python/setup.py | 20 +++++++++++++++----- 1 file changed, 15 insertions(+), 5 deletions(-) diff --git a/sdks/python/setup.py b/sdks/python/setup.py index dbb1dd68f17..933616ca52e 100644 --- a/sdks/python/setup.py +++ b/sdks/python/setup.py @@ -168,8 +168,10 @@ ml_base = [ 'langchain', 'sentence-transformers>=2.2.2', 'skl2onnx', - 'pyod', + 'pyod>=0.7.6', # 0.7.5 crashes setuptools 'tensorflow', + # tensorflow transient dep, lower versions crash install on Python3.10+ + 'absl-py>=0.12.0', 'tensorflow-hub', 'tf2onnx', 'torch', @@ -547,7 +549,7 @@ if __name__ == '__main__': # tests due to tag check introduced since pip 24.2 # https://github.com/apache/beam/issues/31285 # 'xgboost<2.0', # https://github.com/apache/beam/issues/31252 - ] + ml_base + milvus_dependency, + ] + ml_base, 'p312_ml_test': [ 'datatable', ] + ml_base, @@ -584,7 +586,11 @@ if __name__ == '__main__': # For more info, see # https://docs.google.com/document/d/1c84Gc-cZRCfrU8f7kWGsNR2o8oSRjCM-dGHO9KvPWPw/edit?usp=sharing 'torch': ['torch>=1.9.0,<2.8.0'], - 'tensorflow': ['tensorflow>=2.12rc1,<2.21'], + 'tensorflow': [ + 'tensorflow>=2.12rc1,<2.21', + # transient dep, lower versions crash install on Python3.10+ + 'absl-py>=0.12.0' + ], 'transformers': [ 'transformers>=4.28.0,<4.56.0', 'tensorflow>=2.12.0', @@ -593,7 +599,9 @@ if __name__ == '__main__': 'ml_cpu': [ 'tensorflow>=2.12.0', 'torch==2.8.0+cpu', - 'transformers>=4.28.0,<4.56.0' + 'transformers>=4.28.0,<4.56.0', + # transient dep, lower versions crash install on Python3.10+ + 'absl-py>=0.12.0' ], 'redis': ['redis>=5.0.0,<6'], 'tft': [ @@ -610,7 +618,9 @@ if __name__ == '__main__': 'tensorflow==2.11.0', 'tf2onnx==1.13.0', 'skl2onnx==1.13', - 'transformers==4.25.1' + 'transformers==4.25.1', + # transient dep, lower versions crash install on Python3.10+ + 'absl-py>=0.12.0' ], 'xgboost': ['xgboost>=1.6.0,<2.1.3', 'datatable==1.0.0'], 'tensorflow-hub': ['tensorflow-hub>=0.14.0,<0.16.0'],
