Fix D105 checks for SSH provider (apache#38013)

Co-authored-by: mehdi.alizade <[email protected]>
drajguru · Mar 14, 2024 · 6391b1a · 6391b1a
1 parent 18aa85b
commit 6391b1a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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]