diff --git a/airflow/providers/google/cloud/sensors/bigquery.py b/airflow/providers/google/cloud/sensors/bigquery.py index 1214d94579..c1f722a7bb 100644 --- a/airflow/providers/google/cloud/sensors/bigquery.py +++ b/airflow/providers/google/cloud/sensors/bigquery.py @@ -71,4 +71,7 @@ def poke(self, context): hook = BigQueryHook( bigquery_conn_id=self.bigquery_conn_id, delegate_to=self.delegate_to) - return hook.table_exists(self.project_id, self.dataset_id, self.table_id) + return hook.table_exists( + project_id=self.project_id, + dataset_id=self.dataset_id, + table_id=self.table_id) diff --git a/tests/providers/google/cloud/sensors/test_bigquery.py b/tests/providers/google/cloud/sensors/test_bigquery.py index 0c9735e3b2..6f48bde74d 100644 --- a/tests/providers/google/cloud/sensors/test_bigquery.py +++ b/tests/providers/google/cloud/sensors/test_bigquery.py @@ -47,7 +47,7 @@ def test_passing_arguments_to_hook(self, mock_hook): delegate_to=TEST_DELEGATE_TO ) mock_hook.return_value.table_exists.assert_called_once_with( - TEST_PROJECT_ID, - TEST_DATASET_ID, - TEST_TABLE_ID + project_id=TEST_PROJECT_ID, + dataset_id=TEST_DATASET_ID, + table_id=TEST_TABLE_ID )