Fix Milvus ML tests

Author: aIbrahiim

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

@@ -34,6 +34,7 @@
 from apache_beam.ml.rag.utils import retry_with_backoff
 from apache_beam.ml.rag.utils import unpack_dataclass_with_kwargs
 from apache_beam.testing.test_pipeline import TestPipeline
+
 # pylint: disable=wrong-import-order, wrong-import-position, ungrouped-imports
 try:
   from pymilvus import CollectionSchema

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

@@ -87,7 +87,8 @@ class CustomMilvusContainer(MilvusContainer):
   Extends MilvusContainer to provide custom port binding and environment
   configuration for testing with standalone Milvus instances.
   """
-  def __init__(
+
+  def __init__(  # pylint: disable=bad-super-call
       self,
       image: str,
       service_container_port,
@@ -96,7 +97,11 @@ def __init__(
   ) -> None:
     # Skip the parent class's constructor and go straight to
     # GenericContainer.
-    super(MilvusContainer, self).__init__(image=image, **kwargs)
+    super(
+        MilvusContainer,
+        self,
+    ).__init__(
+        image=image, **kwargs)
     self.port = service_container_port
     self.healthcheck_port = healthcheck_container_port
     self.with_exposed_ports(service_container_port, healthcheck_container_port)
@@ -135,8 +140,10 @@ class MilvusTestHelpers:
   # 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 reports started before gRPC is ready."""
+    """Wait until Milvus accepts RPCs.
 
+    Docker may report started before gRPC is ready.
+    """
     def list_collections_probe():
       client = MilvusClient(uri=uri)
       try: