refactor: Update dremio_source.py

sagar-salvi-apptware · Oct 11, 2024 · 84227b6 · 84227b6
1 parent 51c2443
commit 84227b6
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_source.py b/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_source.py
@@ -7,6 +7,7 @@
 from datetime import datetime
 from typing import Any, Dict, Iterable, List, Optional
 
+import datahub.sql_parsing.sqlglot_utils
 from datahub.emitter.mce_builder import (
     make_data_platform_urn,
     make_dataset_urn_with_platform_instance,
@@ -606,3 +607,10 @@ def close(self) -> None:
         Close any resources held by the source.
         """
         pass
+
+
+def _sql_dialect(platform: str) -> str:
+    return "trino"
+
+
+datahub.sql_parsing.sqlglot_utils._get_dialect_str = _sql_dialect