fix(snowflake): Fixing top query trimming in snowflake (datahub-proje…

…ct#6447)
CybercentreCanada · Feb 1, 2023 · 1b0d483 · 1b0d483
1 parent b6347ee
commit 1b0d483
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/snowflake/snowflake_usage_v2.py b/metadata-ingestion/src/datahub/ingestion/source/snowflake/snowflake_usage_v2.py
@@ -194,7 +194,7 @@ def _map_top_sql_queries(self, top_sql_queries: Dict) -> List[str]:
             [
                 trim_query(format_sql_query(query), budget_per_query)
                 if self.config.format_sql_queries
-                else query
+                else trim_query(query, budget_per_query)
                 for query in top_sql_queries
             ]
         )