fix: unambiguous trino driver

apache · Feb 11, 2022 · 565c59d · 565c59d
1 parent a7d505d
commit 565c59d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -62,7 +62,10 @@ def get_git_sha() -> str:
     packages=find_packages(),
     include_package_data=True,
     zip_safe=False,
-    entry_points={"console_scripts": ["superset=superset.cli.main:superset"]},
+    entry_points={
+        "console_scripts": ["superset=superset.cli.main:superset"],
+        "sqlalchemy.dialects": ["trinonative = sqlalchemy_trino.dialect:TrinoDialect"],
+    },
     install_requires=[
         "backoff>=1.8.0",
         "bleach>=3.0.2, <4.0.0",

diff --git a/superset/db_engine_specs/trino.py b/superset/db_engine_specs/trino.py
@@ -34,6 +34,7 @@
 
 class TrinoEngineSpec(BaseEngineSpec):
     engine = "trino"
+    engine_aliases = {"trinonative"}
     engine_name = "Trino"
 
     _time_grain_expressions = {