fix bq_to_mysql init checks (#37653)

apache · Feb 23, 2024 · 8c05e59 · 8c05e59
1 parent 310a427
commit 8c05e59
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -327,7 +327,6 @@ repos:
               ^.*__init__\.py$|
               ^airflow\/providers\/amazon\/aws\/transfers\/gcs_to_s3\.py$|
               ^airflow\/providers\/databricks\/operators\/databricks\.py$|
-              ^airflow\/providers\/google\/cloud\/transfers\/bigquery_to_mysql\.py$|
               ^airflow\/providers\/amazon\/aws\/transfers\/redshift_to_s3\.py$|
               ^airflow\/providers\/google\/cloud\/operators\/compute\.py$|
               ^airflow\/providers\/amazon\/aws\/operators\/emr\.py$|

diff --git a/airflow/providers/google/cloud/transfers/bigquery_to_mysql.py b/airflow/providers/google/cloud/transfers/bigquery_to_mysql.py
@@ -48,6 +48,8 @@ def __init__(
         mysql_table: str | None = None,
         target_table_name: str | None = None,
         mysql_conn_id: str = "mysql_default",
+        dataset_id: str | None = None,
+        table_id: str | None = None,
         **kwargs,
     ) -> None:
         if mysql_table is not None:
@@ -65,7 +67,9 @@ def __init__(
 
             target_table_name = mysql_table
 
-        super().__init__(target_table_name=target_table_name, **kwargs)
+        super().__init__(
+            target_table_name=target_table_name, dataset_id=dataset_id, table_id=table_id, **kwargs
+        )
         self.mysql_conn_id = mysql_conn_id
 
     def get_sql_hook(self) -> MySqlHook: