throw errors if key_filename is not found (#383)

When user provides key_filename and it does not exist, it's expected to throw an error. --------- Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
deepmodeling · Oct 20, 2023 · 3ba5626 · 3ba5626
1 parent 7b5382e
commit 3ba5626
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dpdispatcher/ssh_context.py b/dpdispatcher/ssh_context.py
@@ -168,6 +168,8 @@ def _setup_ssh(self):
                         pass
                     if key is not None:
                         break
+            else:
+                raise OSError(f"{key_path} not found!")
         elif self.look_for_keys:
             for keytype, name in [
                 (paramiko.RSAKey, "rsa"),