From bdb3e488e4795953eca16e1b682429759a1cdd31 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Tue, 21 Jan 2025 11:06:10 +0100
Subject: [PATCH] Trying to avoid the random timeout. (#2929)

* Trying to avoid the random timeout.

* More read timeout ?

* Longer timeout ?

* Remove legacy ENV directive.

* Remove the dummy test, only increase the read timeout.

* Wat?
---
 Dockerfile                    | 2 +-
 integration-tests/conftest.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 18b1d5b0..f7236e72 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -58,7 +58,7 @@ ARG INSTALL_CHANNEL=pytorch
 # Automatically set by buildx
 ARG TARGETPLATFORM
 
-ENV PATH /opt/conda/bin:$PATH
+ENV PATH=/opt/conda/bin:$PATH
 
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
         build-essential \
diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py
index c702ae70..6f8aa715 100644
--- a/integration-tests/conftest.py
+++ b/integration-tests/conftest.py
@@ -562,6 +562,7 @@ def launcher(event_loop):
                 docker.types.DeviceRequest(count=gpu_count, capabilities=[["gpu"]])
             ]
 
+        client.api.timeout = 1000
         container = client.containers.run(
             DOCKER_IMAGE,
             command=args,
@@ -573,7 +574,7 @@ def launcher(event_loop):
             devices=devices,
             volumes=volumes,
             ports={"80/tcp": port},
-            healthcheck={"timeout": int(60 * 1e9), "retries": 2},  # 60s
+            healthcheck={"timeout": int(180 * 1e9), "retries": 2},  # 60s
             shm_size="1G",
         )