From 9c7abeba32ba33f96ae6fcd928332f1d9f2c04e7 Mon Sep 17 00:00:00 2001
From: Adrien <adrien@huggingface.co>
Date: Thu, 18 Jul 2024 18:16:13 +0200
Subject: [PATCH] ww

Signed-off-by: Adrien <adrien@huggingface.co>
---
 integration-tests/conftest.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py
index 60146ad1..1c9cc8c9 100644
--- a/integration-tests/conftest.py
+++ b/integration-tests/conftest.py
@@ -466,7 +466,8 @@ def launcher(event_loop):
             args.append("--cuda-graphs")
             args.append(",".join(map(str, cuda_graphs)))
 
-        client = docker.from_env()
+        client = docker.from_env(timeout=180)
+        client.ping()
 
         container_name = f"tgi-tests-{model_id.split('/')[-1]}-{num_shard}-{quantize}"