From 84227b6091212fe32654a4b7ff8f1a7883daf53d Mon Sep 17 00:00:00 2001 From: sagar-salvi-apptware Date: Fri, 11 Oct 2024 18:54:35 +0530 Subject: [PATCH] refactor: Update dremio_source.py --- .../src/datahub/ingestion/source/dremio/dremio_source.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_source.py b/metadata-ingestion/src/datahub/ingestion/source/dremio/dremio_source.py index cd5c91cbb5d1e3..ddac5bf3a3e30c 100644 --- 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