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: