From d43e883e2030f72d07ec7c2080e2aaee8226e214 Mon Sep 17 00:00:00 2001 From: Hemil Desai Date: Tue, 10 Jun 2025 15:42:32 -0700 Subject: [PATCH 1/2] Configurable docker client timeout Signed-off-by: Hemil Desai --- nemo_run/core/execution/docker.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nemo_run/core/execution/docker.py b/nemo_run/core/execution/docker.py index 4de7433d..9085bffa 100644 --- a/nemo_run/core/execution/docker.py +++ b/nemo_run/core/execution/docker.py @@ -38,6 +38,7 @@ from nemo_run.core.serialization.yaml import YamlSerializer from nemo_run.core.serialization.zlib_json import ZlibJSONSerializer + if TYPE_CHECKING: from docker import DockerClient from docker.models.containers import Container @@ -63,7 +64,7 @@ def get_client() -> "DockerClient": import docker - return docker.from_env() + return docker.from_env(timeout=int(os.environ.get("NEMO_RUN_DOCKER_CLIENT_TIMEOUT", 60))) def ensure_network(client: Optional["DockerClient"] = None, network: Optional[str] = None) -> None: From 06dd2e01c9aba8a19425e76c95eb29f63771e618 Mon Sep 17 00:00:00 2001 From: Hemil Desai Date: Tue, 10 Jun 2025 15:44:22 -0700 Subject: [PATCH 2/2] fix Signed-off-by: Hemil Desai --- nemo_run/core/execution/docker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nemo_run/core/execution/docker.py b/nemo_run/core/execution/docker.py index 9085bffa..4799e279 100644 --- a/nemo_run/core/execution/docker.py +++ b/nemo_run/core/execution/docker.py @@ -64,7 +64,7 @@ def get_client() -> "DockerClient": import docker - return docker.from_env(timeout=int(os.environ.get("NEMO_RUN_DOCKER_CLIENT_TIMEOUT", 60))) + return docker.from_env(timeout=int(os.environ.get("NEMORUN_DOCKER_CLIENT_TIMEOUT", 60))) def ensure_network(client: Optional["DockerClient"] = None, network: Optional[str] = None) -> None: