From 8c05e59922946ee06bc014d578676581b33dba2c Mon Sep 17 00:00:00 2001 From: Albert Olweny <41315252+okirialbert@users.noreply.github.com> Date: Fri, 23 Feb 2024 21:14:11 +0300 Subject: [PATCH] fix bq_to_mysql init checks (#37653) --- .pre-commit-config.yaml | 1 - .../providers/google/cloud/transfers/bigquery_to_mysql.py | 6 +++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 8fcfd4c431c9a..b0e1a27700884 100644 --- 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 index 166305f76629b..66084a6fb3ca3 100644 --- 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: