From 93f5834c8a0670d34079233537e223f7e20f9297 Mon Sep 17 00:00:00 2001 From: LanderOtto Date: Wed, 25 Sep 2024 11:27:48 +0200 Subject: [PATCH] Possible solution --- streamflow/deployment/connector/ssh.py | 1 + streamflow/deployment/stream.py | 1 + 2 files changed, 2 insertions(+) diff --git a/streamflow/deployment/connector/ssh.py b/streamflow/deployment/connector/ssh.py index f466b17f3..00e9f88aa 100644 --- a/streamflow/deployment/connector/ssh.py +++ b/streamflow/deployment/connector/ssh.py @@ -204,6 +204,7 @@ async def __aenter__(self) -> asyncssh.SSHClientProcess: async def __aexit__(self, exc_type, exc_val, exc_tb): async with self._condition: if self._selected_context: + await self._selected_context.close() if self._proc: await self._proc.__aexit__(exc_type, exc_val, exc_tb) self._condition.notify_all() diff --git a/streamflow/deployment/stream.py b/streamflow/deployment/stream.py index dfccf1ba5..26ea8359b 100644 --- a/streamflow/deployment/stream.py +++ b/streamflow/deployment/stream.py @@ -38,6 +38,7 @@ async def write(self, data: Any): class StreamWriterWrapper(StreamWrapper): async def close(self): + self.stream.write_eof() self.stream.close() await self.stream.wait_closed()