Skip to content

Commit 39394c2

Browse files
committed
Fix ml_base dependency
1 parent 5c2c353 commit 39394c2

1 file changed

Lines changed: 10 additions & 4 deletions

File tree

sdks/python/setup.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -168,8 +168,9 @@ def cythonize(*args, **kwargs):
168168
'langchain',
169169
'sentence-transformers>=2.2.2',
170170
'skl2onnx',
171-
'pyod',
171+
'pyod>=0.7.6', # 0.7.5 crashes setuptools
172172
'tensorflow',
173+
'absl-py>=0.12.0', # tensorflow transient dependency, lower versions not compatible with Python3.10+
173174
'tensorflow-hub',
174175
'tf2onnx',
175176
'torch',
@@ -584,7 +585,10 @@ def get_portability_package_data():
584585
# For more info, see
585586
# https://docs.google.com/document/d/1c84Gc-cZRCfrU8f7kWGsNR2o8oSRjCM-dGHO9KvPWPw/edit?usp=sharing
586587
'torch': ['torch>=1.9.0,<2.8.0'],
587-
'tensorflow': ['tensorflow>=2.12rc1,<2.21'],
588+
'tensorflow': [
589+
'tensorflow>=2.12rc1,<2.21',
590+
'absl-py>=0.12.0' # tensorflow transient dependency, lower versions not compatible with Python3.10+
591+
],
588592
'transformers': [
589593
'transformers>=4.28.0,<4.56.0',
590594
'tensorflow>=2.12.0',
@@ -593,7 +597,8 @@ def get_portability_package_data():
593597
'ml_cpu': [
594598
'tensorflow>=2.12.0',
595599
'torch==2.8.0+cpu',
596-
'transformers>=4.28.0,<4.56.0'
600+
'transformers>=4.28.0,<4.56.0',
601+
'absl-py>=0.12.0' # tensorflow transient dependency, lower versions not compatible with Python3.10+
597602
],
598603
'redis': ['redis>=5.0.0,<6'],
599604
'tft': [
@@ -610,7 +615,8 @@ def get_portability_package_data():
610615
'tensorflow==2.11.0',
611616
'tf2onnx==1.13.0',
612617
'skl2onnx==1.13',
613-
'transformers==4.25.1'
618+
'transformers==4.25.1',
619+
'absl-py>=0.12.0' # tensorflow transient dependency, lower versions not compatible with Python3.10+
614620
],
615621
'xgboost': ['xgboost>=1.6.0,<2.1.3', 'datatable==1.0.0'],
616622
'tensorflow-hub': ['tensorflow-hub>=0.14.0,<0.16.0'],

0 commit comments

Comments
 (0)