From ebb95b8a9aab9fad90950810010fa5b61d1bd976 Mon Sep 17 00:00:00 2001 From: Harshal Sheth Date: Tue, 21 May 2024 22:38:36 -0700 Subject: [PATCH] fix(ingest/airflow): fix support for bigquery insert job operator (#10567) --- .../airflow-plugin/src/datahub_airflow_plugin/_extractors.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metadata-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/_extractors.py b/metadata-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/_extractors.py index f91c77591d35b..731c9141c4cde 100644 --- a/metadata-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/_extractors.py +++ b/metadata-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/_extractors.py @@ -203,7 +203,7 @@ def _parse_sql_into_task_metadata( task_name = f"{self.operator.dag_id}.{self.operator.task_id}" run_facets = {} - job_facets = {"sql": SqlJobFacet(query=self._normalize_sql(sql))} + job_facets = {"sql": SqlJobFacet(query=SqlExtractor._normalize_sql(sql))} # Prepare to run the SQL parser. graph = self.context.get(_DATAHUB_GRAPH_CONTEXT_KEY, None)