diff --git a/airflow/providers/ssh/hooks/ssh.py b/airflow/providers/ssh/hooks/ssh.py index 5ed59622a70cdc..788f5f3f0aede6 100644 --- a/airflow/providers/ssh/hooks/ssh.py +++ b/airflow/providers/ssh/hooks/ssh.py @@ -375,9 +375,11 @@ def log_before_sleep(retry_state): category=AirflowProviderDeprecationWarning, ) def __enter__(self) -> SSHHook: + """Return an instance of SSHHook when the `with` statement is used.""" return self def __exit__(self, exc_type, exc_val, exc_tb) -> None: + """Clear ssh client after exiting the `with` statement block.""" if self.client is not None: self.client.close() self.client = None diff --git a/pyproject.toml b/pyproject.toml index 880731a4c9e499..cd786e93644481 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1470,7 +1470,6 @@ combine-as-imports = true "airflow/providers/microsoft/psrp/hooks/psrp.py" = ["D105"] "airflow/providers/samba/hooks/samba.py" = ["D105"] "airflow/providers/smtp/hooks/smtp.py" = ["D105"] -"airflow/providers/ssh/hooks/ssh.py" = ["D105"] "airflow/providers/tableau/hooks/tableau.py" = ["D105"] [tool.ruff.lint.flake8-tidy-imports]