Skip to content

Commit 1c1e9ad

Browse files
authored
fix: properly share engine adapter config to spark engine adapter (#4162)
1 parent d1beef6 commit 1c1e9ad

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

sqlmesh/core/engine_adapter/databricks.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ class DatabricksEngineAdapter(SparkEngineAdapter):
4545

4646
def __init__(self, *args: t.Any, **kwargs: t.Any) -> None:
4747
super().__init__(*args, **kwargs)
48-
self._set_spark_engine_adapter_if_needed()
48+
self._set_spark_engine_adapter_if_needed(kwargs.get("multithreaded", False))
4949

5050
@classmethod
5151
def can_access_spark_session(cls, disable_spark_session: bool) -> bool:
@@ -92,7 +92,7 @@ def _use_spark_session(self) -> bool:
9292
def is_spark_session_connection(self) -> bool:
9393
return isinstance(self.connection, SparkSessionConnection)
9494

95-
def _set_spark_engine_adapter_if_needed(self) -> None:
95+
def _set_spark_engine_adapter_if_needed(self, multithreaded: bool) -> None:
9696
self._spark_engine_adapter = None
9797

9898
if not self._use_spark_session or self.is_spark_session_connection:
@@ -117,6 +117,11 @@ def _set_spark_engine_adapter_if_needed(self) -> None:
117117
partial(connection, spark=spark, catalog=catalog),
118118
default_catalog=catalog,
119119
execute_log_level=self._execute_log_level,
120+
multithreaded=multithreaded,
121+
sql_gen_kwargs=self._sql_gen_kwargs,
122+
register_comments=self._register_comments,
123+
pre_ping=self._pre_ping,
124+
pretty_sql=self._pretty_sql,
120125
)
121126

122127
@property

0 commit comments

Comments
 (0)