Fix: Import validation for the snowflake connection config (#4723)

Author: izeigerman

sqlmesh/core/config/connection.py

@@ -539,7 +539,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:
@@ -566,6 +565,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]:
         """
@@ -733,7 +736,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:
@@ -811,6 +813,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: