Fix Milvus ML PreCommit xdist, container startup, and grpc readiness

Author: aIbrahiim

.github/trigger_files/beam_PreCommit_Python_ML.json

@@ -1,4 +1,4 @@
 {
   "comment": "Modify this file in a trivial way to cause this test suite to run",
-  "revision": 3
+  "revision": 6
 }

sdks/python/apache_beam/ml/rag/enrichment/milvus_search_it_test.py

@@ -225,6 +225,7 @@ def __getitem__(self, key):
 
 
 @pytest.mark.require_docker_in_docker
+@pytest.mark.no_xdist
 @unittest.skipUnless(
     platform.system() == "Linux",
     "Test runs only on Linux due to lack of support, as yet, for nested "

sdks/python/apache_beam/ml/rag/ingestion/milvus_search_it_test.py

@@ -162,6 +162,7 @@ def drop_collection(client: MilvusClient, collection_name: str):
 
 
 @pytest.mark.require_docker_in_docker
+@pytest.mark.no_xdist
 @unittest.skipIf(not PYMILVUS_AVAILABLE, 'pymilvus is not installed.')
 @unittest.skipUnless(
     platform.system() == "Linux",

sdks/python/apache_beam/ml/rag/test_utils.py

@@ -113,6 +113,7 @@ def __init__(  # pylint: disable=bad-super-call
     self.with_bind_ports(service_container_port, service_host_port)
     self.with_bind_ports(healthcheck_container_port, healthcheck_host_port)
     self.cmd = "milvus run standalone"
+    self.with_command(self.cmd)
 
     # Set environment variables needed for Milvus.
     envs = {
@@ -152,9 +153,8 @@ def list_collections_probe():
 
     retry_with_backoff(
         list_collections_probe,
-        max_retries=25,
+        max_retries=5,
         retry_delay=2.0,
-        retry_backoff_factor=1.2,
         operation_name="Milvus client connection after container start",
         exception_types=(MilvusException, ))