From a9e85b3a5fb2d43ade37f057bb85c7aa651ffa2b Mon Sep 17 00:00:00 2001
From: Magnus Bakken <10287813+magbak@users.noreply.github.com>
Date: Mon, 26 Aug 2024 16:36:47 +0200
Subject: [PATCH] Begin databricks support

---
 lib/virtualization/src/python/sql_translation.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/virtualization/src/python/sql_translation.rs b/lib/virtualization/src/python/sql_translation.rs
index afdf883..7e58940 100644
--- a/lib/virtualization/src/python/sql_translation.rs
+++ b/lib/virtualization/src/python/sql_translation.rs
@@ -26,7 +26,7 @@ with warnings.catch_warnings():
         package = "bq"
         inherit_cache = True
 
-def translate_sql(vq: VirtualizedQuery, dialect: Literal["bigquery", "postgres"],
+def translate_sql(vq: VirtualizedQuery, dialect: Literal["bigquery", "postgres", "databricks"],
                   resource_sql_map: Dict[str, Any]) -> str:
     mapper = SPARQLMapper(dialect, resource_sql_map)
     q = mapper.virtualized_query_to_sql(vq)