Fix The PreCommit Python ML tests with ML deps installed job

.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
 }

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 = {