Fix MyPy errors for google/marketing_platform and suite (#20227)

Part of #19891
apache · Dec 13, 2021 · 644051a · 644051a
1 parent a208463
commit 644051a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/airflow/providers/google/marketing_platform/hooks/display_video.py b/airflow/providers/google/marketing_platform/hooks/display_video.py
@@ -124,7 +124,7 @@ def list_queries(
         response = self.get_conn().queries().listqueries().execute(num_retries=self.num_retries)
         return response.get('queries', [])
 
-    def run_query(self, query_id: str, params: Dict[str, Any]) -> None:
+    def run_query(self, query_id: str, params: Optional[Dict[str, Any]]) -> None:
         """
         Runs a stored query to generate a report.
 

diff --git a/airflow/providers/google/marketing_platform/operators/display_video.py b/airflow/providers/google/marketing_platform/operators/display_video.py
@@ -376,8 +376,8 @@ def __init__(
         self,
         *,
         report_id: str,
-        params: Dict[str, Any] = None,
-        parameters: Dict[str, Any] = None,
+        params: Optional[Dict[str, Any]] = None,
+        parameters: Optional[Dict[str, Any]] = None,
         api_version: str = "v1",
         gcp_conn_id: str = "google_cloud_default",
         delegate_to: Optional[str] = None,

diff --git a/airflow/providers/google/suite/transfers/sql_to_sheets.py b/airflow/providers/google/suite/transfers/sql_to_sheets.py
@@ -77,7 +77,7 @@ def __init__(
         spreadsheet_id: str,
         sql_conn_id: str,
         parameters: Optional[Union[Mapping, Iterable]] = None,
-        database: str = None,
+        database: Optional[str] = None,
         spreadsheet_range: str = "Sheet1",
         gcp_conn_id: str = "google_cloud_default",
         delegate_to: Optional[str] = None,