Fixing top query trimming in snowflake

datahub-project · Nov 15, 2022 · b9f90e1 · b9f90e1
1 parent fc26cf3
commit b9f90e1
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
             ]
         )