From ad8ce24ab6f4706e7a67da33797e689638cd1d11 Mon Sep 17 00:00:00 2001 From: Iaroslav Zeigerman Date: Wed, 11 Jun 2025 12:11:58 -0700 Subject: [PATCH] Fix: Import validation for the snowflake connection config --- sqlmesh/core/config/connection.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/sqlmesh/core/config/connection.py b/sqlmesh/core/config/connection.py index e7e138c908..db38f7b30d 100644 --- a/sqlmesh/core/config/connection.py +++ b/sqlmesh/core/config/connection.py @@ -540,7 +540,6 @@ class SnowflakeConnectionConfig(ConnectionConfig): type_: t.Literal["snowflake"] = Field(alias="type", default="snowflake") _concurrent_tasks_validator = concurrent_tasks_validator - _engine_import_validator = _get_engine_import_validator("snowflake", "snowflake") @model_validator(mode="before") def _validate_authenticator(cls, data: t.Any) -> t.Any: @@ -567,6 +566,10 @@ def _validate_authenticator(cls, data: t.Any) -> t.Any: return data + _engine_import_validator = _get_engine_import_validator( + "snowflake.connector.network", "snowflake" + ) + @classmethod def _get_private_key(cls, values: t.Dict[str, t.Optional[str]], auth: str) -> t.Optional[bytes]: """ @@ -734,7 +737,6 @@ class DatabricksConnectionConfig(ConnectionConfig): _concurrent_tasks_validator = concurrent_tasks_validator _http_headers_validator = http_headers_validator - _engine_import_validator = _get_engine_import_validator("databricks", "databricks") @model_validator(mode="before") def _databricks_connect_validator(cls, data: t.Any) -> t.Any: @@ -812,6 +814,8 @@ def _databricks_connect_validator(cls, data: t.Any) -> t.Any: return data + _engine_import_validator = _get_engine_import_validator("databricks", "databricks") + @property def _connection_kwargs_keys(self) -> t.Set[str]: if self.use_spark_session_only: