From 1d4977cbb7a5427d49f80ac7c34e9e7b526fe356 Mon Sep 17 00:00:00 2001 From: Harshal Sheth Date: Wed, 12 Jun 2024 01:57:36 -0500 Subject: [PATCH] feat(ingest): make query formatting more robust (#10678) --- metadata-ingestion/src/datahub/sql_parsing/sqlglot_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/metadata-ingestion/src/datahub/sql_parsing/sqlglot_utils.py b/metadata-ingestion/src/datahub/sql_parsing/sqlglot_utils.py index b494dde4013a4..f74a915c9fe82 100644 --- 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