From cadbc8dbe039a12739fc3bbfb225928a490421d3 Mon Sep 17 00:00:00 2001 From: mahdi alizade <79321261+alizademhdi@users.noreply.github.com> Date: Mon, 11 Mar 2024 10:00:07 +0330 Subject: [PATCH] Fix D105 checks for SSH provider (#38013) Co-authored-by: mehdi.alizade --- airflow/providers/ssh/hooks/ssh.py | 2 ++ pyproject.toml | 1 - 2 files changed, 2 insertions(+), 1 deletion(-) 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]