Fix grpc setup timeout

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": 4
+  "revision": 5
 }

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

@@ -137,27 +137,6 @@ class MilvusTestHelpers:
   # https://milvus.io/docs/release_notes.md or PyPI at
   # https://pypi.org/project/pymilvus/ for version compatibility.
   # Example: Milvus v2.6.0 requires pymilvus==2.6.0 (exact match required).
-  @staticmethod
-  def _wait_for_milvus_grpc(uri: str) -> None:
-    """Wait until Milvus accepts RPCs.
-
-    Docker may report started before gRPC is ready.
-    """
-    def list_collections_probe():
-      client = MilvusClient(uri=uri)
-      try:
-        client.list_collections()
-      finally:
-        client.close()
-
-    retry_with_backoff(
-        list_collections_probe,
-        max_retries=25,
-        retry_delay=2.0,
-        retry_backoff_factor=1.2,
-        operation_name="Milvus client connection after container start",
-        exception_types=(MilvusException, ))
-
   @staticmethod
   def start_db_container(
       image="milvusdb/milvus:v2.5.10",
@@ -188,7 +167,6 @@ def start_db_container(
           host = running_container.get_container_host_ip()
           port = running_container.get_exposed_port(service_container_port)
           info = VectorDBContainerInfo(running_container, host, port)
-          MilvusTestHelpers._wait_for_milvus_grpc(info.uri)
           _LOGGER.info(
               "milvus db container started successfully on %s.", info.uri)
           break