sdks/python: make milvus as extra dependency

Author: mohamedawnallah

sdks/python/setup.py

@@ -401,7 +401,6 @@ def get_portability_package_data():
           'typing-extensions>=3.7.0',
           'zstandard>=0.18.0,<1',
           'pyyaml>=3.12,<7.0.0',
-          'pymilvus>=2.5.10,<3.0.0',
           # Dynamic dependencies must be specified in a separate list, otherwise
           # Dependabot won't be able to parse the main list. Any dynamic
           # dependencies will not receive updates from Dependabot.
@@ -588,7 +587,8 @@ def get_portability_package_data():
               'transformers==4.25.1'
           ],
           'xgboost': ['xgboost>=1.6.0,<2.1.3', 'datatable==1.0.0'],
-          'tensorflow-hub': ['tensorflow-hub>=0.14.0,<0.16.0']
+          'tensorflow-hub': ['tensorflow-hub>=0.14.0,<0.16.0'],
+          'milvus': ['pymilvus>=2.5.10,<3.0.0']
       },
       zip_safe=False,
       # PyPI package information.

sdks/python/tox.ini

@@ -562,3 +562,10 @@ commands =
   /bin/sh -c "pip freeze | grep -E tensorflow"
   # Allow exit code 5 (no tests run) so that we can run this command safely on arbitrary subdirectories.
   bash {toxinidir}/scripts/run_pytest.sh {envname} 'apache_beam/ml/transforms/embeddings'
+
+[testenv:py{39,310,311,312,313}-milvus]
+extras = test,gcp,milvus
+commands =
+  # Log versions for debugging.
+  /bin/sh -c "pip freeze | grep -E 'pymilvus|testcontainers'"
+  bash {toxinidir}/scripts/run_pytest.sh {envname} "{posargs}"