@@ -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