From 2654811313a6521c4975274909611a6907ebf511 Mon Sep 17 00:00:00 2001 From: David Ankin Date: Tue, 21 Jan 2025 15:07:22 -0500 Subject: [PATCH] fix: milvus healthcheck: use correct requests errors --- modules/milvus/testcontainers/milvus/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/milvus/testcontainers/milvus/__init__.py b/modules/milvus/testcontainers/milvus/__init__.py index 39a1403e9..2a1534146 100644 --- a/modules/milvus/testcontainers/milvus/__init__.py +++ b/modules/milvus/testcontainers/milvus/__init__.py @@ -69,7 +69,7 @@ def _get_healthcheck_url(self) -> str: port = self.get_exposed_port(self.healthcheck_port) return f"http://{ip}:{port}" - @wait_container_is_ready(requests.exceptions.HTTPError) + @wait_container_is_ready(requests.exceptions.HTTPError, requests.exceptions.ConnectionError) def _healthcheck(self) -> None: healthcheck_url = self._get_healthcheck_url() response = requests.get(f"{healthcheck_url}/healthz", timeout=1)