From fdd722e16ecdbad89df59575eab3498994167b82 Mon Sep 17 00:00:00 2001 From: Adrien Date: Thu, 18 Jul 2024 18:23:54 +0200 Subject: [PATCH] wip Signed-off-by: Adrien --- integration-tests/conftest.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py index 1c9cc8c9..60146ad1 100644 --- a/integration-tests/conftest.py +++ b/integration-tests/conftest.py @@ -466,8 +466,7 @@ def launcher(event_loop): args.append("--cuda-graphs") args.append(",".join(map(str, cuda_graphs))) - client = docker.from_env(timeout=180) - client.ping() + client = docker.from_env() container_name = f"tgi-tests-{model_id.split('/')[-1]}-{num_shard}-{quantize}"