diff --git a/tools/environments/persistent_shell.py b/tools/environments/persistent_shell.py index df1a78ef9..4b89db471 100644 --- a/tools/environments/persistent_shell.py +++ b/tools/environments/persistent_shell.py @@ -40,6 +40,7 @@ class PersistentShellMixin: def _cleanup_temp_files(self): ... _session_id: str = "" + _poll_interval: float = 0.01 @property def _temp_prefix(self) -> str: @@ -223,7 +224,7 @@ class PersistentShellMixin: ) self._send_to_shell(ipc_script) deadline = time.monotonic() + timeout - poll_interval = 0.15 + poll_interval = self._poll_interval while True: if is_interrupted(): diff --git a/tools/environments/ssh.py b/tools/environments/ssh.py index c48b38509..90532dda0 100644 --- a/tools/environments/ssh.py +++ b/tools/environments/ssh.py @@ -77,6 +77,8 @@ class SSHEnvironment(PersistentShellMixin, BaseEnvironment): except subprocess.TimeoutExpired: raise RuntimeError(f"SSH connection to {self.user}@{self.host} timed out") + _poll_interval: float = 0.15 + @property def _temp_prefix(self) -> str: return f"/tmp/hermes-ssh-{self._session_id}"