fix(ingest/airflow): fix support for bigquery insert job operator (da…

…tahub-project#10567)
sleeperdeep · Jun 25, 2024 · 92f367d · 92f367d
1 parent 1099f23
commit 92f367d
Showing 1 changed file with 1 addition and 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
@@ -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)