Skip to content

Commit d81c88a

Browse files
committed
fix ml tests
1 parent e88da14 commit d81c88a

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

sdks/python/setup.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,11 @@ def cythonize(*args, **kwargs):
160160
'pandas>=1.4.3,!=1.5.0,!=1.5.1,<2.3',
161161
]
162162

163-
milvus_dependency = ['pymilvus>=2.5.10,<3.0.0']
163+
milvus_dependency = [
164+
'pymilvus>=2.5.10,<3.0.0; python_version >= "3.13"',
165+
'pymilvus>=2.4.0,<2.5.0; python_version <= "3.12"',
166+
'setuptools>=70.0.0',
167+
]
164168

165169
ml_base = [
166170
'embeddings',
@@ -466,7 +470,8 @@ def get_portability_package_data():
466470
'sqlalchemy-pytds>=1.0.2',
467471
'pg8000>=1.31.5',
468472
"PyMySQL>=1.1.0",
469-
'oracledb>=3.1.1'
473+
'oracledb>=3.1.1',
474+
'aiohttp!=4.0.0a1',
470475
] + milvus_dependency,
471476
'gcp': [
472477
'cachetools>=3.1.0,<7',
@@ -508,6 +513,7 @@ def get_portability_package_data():
508513
'keyrings.google-artifactregistry-auth',
509514
'orjson>=3.9.7,<4',
510515
'regex>=2020.6.8',
516+
'aiohttp!=4.0.0a1',
511517
],
512518
'interactive': [
513519
'facets-overview>=1.1.0,<2',

0 commit comments

Comments
 (0)