Skip to content

Commit 7174991

Browse files
authored
fix(python-sdk): restrict transformers version for python 3.9 compat (#36385)
* fix(python-sdk): restrict transformers version for python 3.9 compatibility Versions 4.55.0+ use Python 3.10+ union syntax which causes TypeError on Python 3.9 * build(python): pin sentence-transformers version for python 3.9 compatibility Use sentence-transformers 2.x for Python 3.9 to maintain compatibility with transformers <4.55.0, while allowing newer versions for Python 3.10+ * docs: add Apache license header to test file Add required license header to comply with Apache Software Foundation requirements
1 parent bd77a1e commit 7174991

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

sdks/python/setup.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,7 +166,11 @@ def cythonize(*args, **kwargs):
166166
'embeddings',
167167
'onnxruntime',
168168
'langchain',
169-
'sentence-transformers',
169+
# sentence-transformers 3.0+ requires transformers 4.34+
170+
# which uses Python 3.10+ union syntax
171+
# Use 2.x versions for Python 3.9 compatibility with transformers <4.55.0
172+
'sentence-transformers>=2.2.2,<3.0.0; python_version < "3.10"',
173+
'sentence-transformers>=2.2.2; python_version >= "3.10"',
170174
'skl2onnx',
171175
'pillow',
172176
'pyod',
@@ -581,7 +585,11 @@ def get_portability_package_data():
581585
'torch': ['torch>=1.9.0,<2.8.0'],
582586
'tensorflow': ['tensorflow>=2.12rc1,<2.21'],
583587
'transformers': [
584-
'transformers>=4.28.0,<4.56.0',
588+
# Restrict transformers to <4.55.0 for Python 3.9 compatibility
589+
# Versions 4.55.0+ use Python 3.10+ union syntax (int | None)
590+
# which causes TypeError on Python 3.9
591+
'transformers>=4.28.0,<4.55.0; python_version < "3.10"',
592+
'transformers>=4.28.0,<4.56.0; python_version >= "3.10"',
585593
'tensorflow>=2.12.0',
586594
'torch>=1.9.0'
587595
],

0 commit comments

Comments
 (0)