diff --git a/superset/db_engine_specs/databricks.py b/superset/db_engine_specs/databricks.py index 7574989abf80d2b72352033c9661a4c223b9595b..fe7f9084009aa163dfa545187a1c12cba4d5e96e 100644 --- a/superset/db_engine_specs/databricks.py +++ b/superset/db_engine_specs/databricks.py @@ -353,7 +353,7 @@ class DatabricksDynamicBaseEngineSpec(BasicParametersMixin, DatabricksBaseEngine class DatabricksNativeEngineSpec(DatabricksDynamicBaseEngineSpec): engine = "databricks" - engine_name = "Databricks" + engine_name = "Databricks (legacy)" drivers = {"connector": "Native all-purpose driver"} default_driver = "connector" @@ -485,7 +485,7 @@ class DatabricksNativeEngineSpec(DatabricksDynamicBaseEngineSpec): class DatabricksPythonConnectorEngineSpec(DatabricksDynamicBaseEngineSpec): engine = "databricks" - engine_name = "Databricks Python Connector" + engine_name = "Databricks" default_driver = "databricks-sql-python" drivers = {"databricks-sql-python": "Databricks SQL Python"} diff --git a/tests/unit_tests/db_engine_specs/test_databricks.py b/tests/unit_tests/db_engine_specs/test_databricks.py index 95403780e11b4a494af34a03e541c0b83f1ffabf..a58500200977cea2a73512d06cab4bbcb0d68c6d 100644 --- a/tests/unit_tests/db_engine_specs/test_databricks.py +++ b/tests/unit_tests/db_engine_specs/test_databricks.py @@ -187,7 +187,7 @@ def test_extract_errors() -> None: error_type=SupersetErrorType.GENERIC_DB_ENGINE_ERROR, level=ErrorLevel.ERROR, extra={ - "engine_name": "Databricks", + "engine_name": "Databricks (legacy)", "issue_codes": [ { "code": 1002, @@ -214,7 +214,7 @@ def test_extract_errors_with_context() -> None: error_type=SupersetErrorType.GENERIC_DB_ENGINE_ERROR, level=ErrorLevel.ERROR, extra={ - "engine_name": "Databricks", + "engine_name": "Databricks (legacy)", "issue_codes": [ { "code": 1002,