pass timeout to paramiko's Transport (#399)

This PR will enable the usage of the "timeout" keyword and support dpdispatcher for machines which require a long time to connect.
deepmodeling · Oct 25, 2023 · 538214f · 538214f
1 parent 238d311
commit 538214f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dpdispatcher/ssh_context.py b/dpdispatcher/ssh_context.py
@@ -140,6 +140,7 @@ def _setup_ssh(self):
         # Make a Paramiko Transport object using the socket
         ts = paramiko.Transport(sock)
         ts.banner_timeout = 60
+        ts.auth_timeout = self.timeout + 20
         ts.use_compression(compress=True)
 
         # Tell Paramiko that the Transport is going to be used as a client