From 4f9c130feceeb7baad2f643d93948cbbb77ed473 Mon Sep 17 00:00:00 2001 From: Kilian Lieret Date: Tue, 12 Nov 2024 11:43:38 -0500 Subject: [PATCH] Enh: Avoid "startint at port None" messages (#141) --- src/swerex/deployment/docker.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/swerex/deployment/docker.py b/src/swerex/deployment/docker.py index 8d3e9e9..b9060e7 100644 --- a/src/swerex/deployment/docker.py +++ b/src/swerex/deployment/docker.py @@ -100,7 +100,8 @@ def _get_swerex_start_cmd(self, token: str) -> list[str]: async def start(self): """Starts the runtime.""" - port = self._port or find_free_port() + if self._port is None: + self._port = find_free_port() assert self._container_name is None self._container_name = self._get_container_name() token = self._get_token() @@ -109,7 +110,7 @@ async def start(self): "run", "--rm", "-p", - f"{port}:8000", + f"{self._port}:8000", *self._docker_args, "--name", self._container_name, @@ -124,7 +125,7 @@ async def start(self): # shell=True required for && etc. self._container_process = subprocess.Popen(cmds, stdout=subprocess.PIPE, stderr=subprocess.PIPE) self.logger.info(f"Starting runtime at {self._port}") - self._runtime = RemoteRuntime(port=port, timeout=self._runtime_timeout, auth_token=token) + self._runtime = RemoteRuntime(port=self._port, timeout=self._runtime_timeout, auth_token=token) t0 = time.time() await self._wait_until_alive(timeout=self._startup_timeout) self.logger.info(f"Runtime started in {time.time() - t0:.2f}s")