ssh: catch EOFError (#420)

Fix #417.
deepmodeling · Dec 13, 2023 · 123b74b · 123b74b
1 parent 0b5ea39
commit 123b74b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dpdispatcher/contexts/ssh_context.py b/dpdispatcher/contexts/ssh_context.py
@@ -295,7 +295,7 @@ def exec_command(self, cmd):
         assert self.ssh is not None
         try:
             return self.ssh.exec_command(cmd)
-        except (paramiko.ssh_exception.SSHException, socket.timeout) as e:
+        except (paramiko.ssh_exception.SSHException, socket.timeout, EOFError) as e:
             # SSH session not active
             # retry for up to 3 times
             # ensure alive