update

Author: mohamedawnallah

sdks/python/setup.py

@@ -522,6 +522,7 @@ def get_portability_package_data():
               'tf2onnx',
               'torch',
               'transformers',
+              'milvus',
               # Comment out xgboost as it is breaking presubmit python ml
               # tests due to tag check introduced since pip 24.2
               # https://github.com/apache/beam/issues/31285

sdks/python/tox.ini

@@ -109,7 +109,7 @@ deps =
   pip==25.0.1
   accelerate>=1.6.0
 setenv =
-extras = test,gcp,dataframe,milvus,ml_test
+extras = test,gcp,dataframe,ml_test
 commands =
   # Log tensorflow version for debugging
   /bin/sh -c "pip freeze | grep -E tensorflow"