From fc821b98f0da395e326a1bc2013a4a4d5a557ff0 Mon Sep 17 00:00:00 2001 From: Harshal Sheth Date: Tue, 21 May 2024 21:39:40 -0700 Subject: [PATCH] fix(ingest/airflow): fix support for bigquery insert job operator --- .../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)