sq

Signed-off-by: Egor Lazarchuk <yegorlz@amazon.co.uk>

tests/host_tools/network.py

@@ -14,13 +14,6 @@
 from framework import utils
 
 
-class SSHConnectionException(Exception):
-    """
-    Specific exception for ssh errors
-    """
-    pass
-
-
 class SSHConnection:
     """
     SSHConnection encapsulates functionality for microVM SSH interaction.
@@ -97,7 +90,7 @@ def scp_get(self, remote_path, local_path, recursive=False):
         self._scp(self.remote_path(remote_path), local_path, opts)
 
     @retry(
-        retry=retry_if_exception_type(SSHConnectionException),
+        retry=retry_if_exception_type(ChildProcessError),
         wait=wait_fixed(0.5),
         stop=stop_after_attempt(20),
         reraise=True,