Fix mypy

astronomer · Dec 8, 2023 · 70f2f35 · 70f2f35
1 parent 24d7be4
commit 70f2f35
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/astronomer/providers/google/cloud/hooks/dataproc.py b/astronomer/providers/google/cloud/hooks/dataproc.py
@@ -3,9 +3,8 @@
 
 from airflow.providers.google.common.consts import CLIENT_INFO
 from airflow.providers.google.common.hooks.base_google import GoogleBaseHook
-from google.api_core import gapic_v1
+from google.api_core import gapic_v1, retry_async as retries
 from google.api_core.client_options import ClientOptions
-from google.api_core import retry_async as retries
 from google.cloud.dataproc_v1 import (
     ClusterControllerAsyncClient,
     Job,
@@ -16,7 +15,7 @@
 try:
     OptionalRetry = Union[retries.AsyncRetry, gapic_v1.method._MethodDefault]
 except AttributeError:
-    OptionalRetry = Union[retries.AsyncRetry, object]  # type: ignore
+    OptionalRetry = Union[retries.AsyncRetry, object]
 
 JobType = Union[Job, Any]
 

diff --git a/astronomer/providers/snowflake/hooks/snowflake_sql_api.py b/astronomer/providers/snowflake/hooks/snowflake_sql_api.py
@@ -143,7 +143,7 @@ def execute_query(
             if e.response is None:
                 raise e
             raise AirflowException(
-                f"Response: {e.response.content}, " f"Status Code: {e.response.status_code}"
+                f"Response: {e.response.content!r}, " f"Status Code: {e.response.status_code}"
             )  # pragma: no cover
         json_response = response.json()
         self.log.info("Snowflake SQL POST API response: %s", json_response)
@@ -208,7 +208,7 @@ def check_query_output(self, query_ids: list[str]) -> None:
                 if e.response is None:
                     raise e
                 raise AirflowException(
-                    f"Response: {e.response.content}, Status Code: {e.response.status_code}"
+                    f"Response: {e.response.content!r}, Status Code: {e.response.status_code}"
                 )
 
     @staticmethod