Possible solution

alpha-unito · Oct 4, 2024 · 93f5834 · 93f5834
1 parent 077abc6
commit 93f5834
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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()