feat(ingest): make query formatting more robust (datahub-project#10678)

acryldata · Jun 12, 2024 · 1d4977c · 1d4977c
1 parent 05aee03
commit 1d4977c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/metadata-ingestion/src/datahub/sql_parsing/sqlglot_utils.py b/metadata-ingestion/src/datahub/sql_parsing/sqlglot_utils.py
@@ -315,8 +315,8 @@ def try_format_query(
 
     try:
         dialect = get_dialect(platform)
-        expression = parse_statement(expression, dialect=dialect)
-        return expression.sql(dialect=dialect, pretty=True)
+        parsed_expression = parse_statement(expression, dialect=dialect)
+        return parsed_expression.sql(dialect=dialect, pretty=True)
     except Exception as e:
         if raises:
             raise