Skip to content

Commit 31b44d4

Browse files
committed
Add _arrow_pandas_type_override and _arrow_to_pandas_kwargs to Connection class
1 parent 647ed39 commit 31b44d4

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/databricks/sql/client.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -234,6 +234,10 @@ def read(self) -> Optional[OAuthToken]:
234234
self.port = kwargs.get("_port", 443)
235235
self.disable_pandas = kwargs.get("_disable_pandas", False)
236236
self.lz4_compression = kwargs.get("enable_query_result_lz4_compression", True)
237+
self._arrow_pandas_type_override = kwargs.get(
238+
"_arrow_pandas_type_override", {}
239+
)
240+
self._arrow_to_pandas_kwargs = kwargs.get("_arrow_to_pandas_kwargs", {})
237241

238242
auth_provider = get_python_sql_connector_auth_provider(
239243
server_hostname, **kwargs

0 commit comments

Comments
 (0)