From cb14cb4b8fc1a81b411bf9c74eea7aee687a3892 Mon Sep 17 00:00:00 2001 From: Luca Soto De Graeve Date: Thu, 12 Dec 2024 16:22:13 +0100 Subject: [PATCH] Auto add ssh missing host keys --- prefect_meemoo/ssh/credentials.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/prefect_meemoo/ssh/credentials.py b/prefect_meemoo/ssh/credentials.py index 1b80744..c1a8aa3 100644 --- a/prefect_meemoo/ssh/credentials.py +++ b/prefect_meemoo/ssh/credentials.py @@ -5,7 +5,7 @@ from prefect import get_run_logger from pydantic import SecretStr from pydantic import Field -from paramiko import SSHClient +from paramiko import SSHClient, AutoAddPolicy class SSHCredentials(Block): @@ -56,7 +56,7 @@ def get_client(self) -> SSHClient: """ client = SSHClient() - client.load_system_host_keys() + client.set_missing_host_key_policy(AutoAddPolicy()) try: client.connect( hostname=self.hostname,