(beam) branch master updated: Fix Milvus ML PreCommit xdist, container startup (#39007)

[yhu]

This is an automated email from the ASF dual-hosted git repository.

Abacn pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/master by this push:
     new 09c962c4aef Fix Milvus ML PreCommit xdist, container startup (#39007)
09c962c4aef is described below

commit 09c962c4aef8a2c21db2f8f85eb8c085f1b28dc0
Author: Abdelrahman Ibrahim <abdoibrahim1...@gmail.com>
AuthorDate: Thu Jun 18 18:24:06 2026 +0200

    Fix Milvus ML PreCommit xdist, container startup (#39007)
---
 .github/trigger_files/beam_PreCommit_Python_ML.json                | 2 +-
 sdks/python/apache_beam/ml/rag/enrichment/milvus_search_it_test.py | 1 +
 sdks/python/apache_beam/ml/rag/ingestion/milvus_search_it_test.py  | 1 +
 sdks/python/apache_beam/ml/rag/test_utils.py                       | 1 +
 4 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/.github/trigger_files/beam_PreCommit_Python_ML.json 
b/.github/trigger_files/beam_PreCommit_Python_ML.json
index 8ed972c9f57..86bf1193abd 100644
--- a/.github/trigger_files/beam_PreCommit_Python_ML.json
+++ b/.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": 7
 }
diff --git a/sdks/python/apache_beam/ml/rag/enrichment/milvus_search_it_test.py 
b/sdks/python/apache_beam/ml/rag/enrichment/milvus_search_it_test.py
index f37fc493148..61d94a69e00 100644
--- a/sdks/python/apache_beam/ml/rag/enrichment/milvus_search_it_test.py
+++ b/sdks/python/apache_beam/ml/rag/enrichment/milvus_search_it_test.py
@@ -225,6 +225,7 @@ MILVUS_IT_CONFIG = {
 
 
 @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 "
diff --git a/sdks/python/apache_beam/ml/rag/ingestion/milvus_search_it_test.py 
b/sdks/python/apache_beam/ml/rag/ingestion/milvus_search_it_test.py
index eec1ee678fa..21b7bb67c9a 100644
--- a/sdks/python/apache_beam/ml/rag/ingestion/milvus_search_it_test.py
+++ b/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",
diff --git a/sdks/python/apache_beam/ml/rag/test_utils.py 
b/sdks/python/apache_beam/ml/rag/test_utils.py
index 72f7bde5d80..b3628b01b2d 100644
--- a/sdks/python/apache_beam/ml/rag/test_utils.py
+++ b/sdks/python/apache_beam/ml/rag/test_utils.py
@@ -113,6 +113,7 @@ class CustomMilvusContainer(MilvusContainer):
     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 = {