fix: milvus healthcheck: use correct requests errors (#759)

testcontainers · Jan 21, 2025 · 78b137c · 78b137c
1 parent ace2a7d
commit 78b137c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)