Skip to content

Commit 5bbfba4

Browse files
upgrade pyarrow and comment lancedb
1 parent e3d8f95 commit 5bbfba4

File tree

4 files changed

+161
-236
lines changed

4 files changed

+161
-236
lines changed

dlt/destinations/impl/databricks/databricks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,7 @@ def __init__(
268268
)
269269
super().__init__(schema, config, sql_client)
270270
self.config: DatabricksClientConfiguration = config
271-
self.sql_client: DatabricksSqlClient = sql_client # type: ignore[assignment]
271+
self.sql_client: DatabricksSqlClient = sql_client
272272
self.type_mapper = DatabricksTypeMapper(self.capabilities)
273273

274274
def start_file_load(self, table: TTableSchema, file_path: str, load_id: str) -> LoadJob:

dlt/destinations/impl/databricks/sql_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828
class DatabricksCursorImpl(DBApiCursorImpl):
2929
"""Use native data frame support if available"""
3030

31-
native_cursor: DatabricksSqlCursor # type: ignore[assignment]
31+
native_cursor: DatabricksSqlCursor
3232
vector_size: ClassVar[int] = 2048
3333

3434
def df(self, chunk_size: int = None, **kwargs: Any) -> DataFrame:
@@ -127,7 +127,7 @@ def execute_query(self, query: AnyStr, *args: Any, **kwargs: Any) -> Iterator[DB
127127
# db_args = kwargs or None
128128

129129
db_args = args or kwargs or None
130-
with self._conn.cursor() as curr: # type: ignore[assignment]
130+
with self._conn.cursor() as curr:
131131
curr.execute(query, db_args)
132132
yield DatabricksCursorImpl(curr) # type: ignore[abstract]
133133

0 commit comments

Comments
 (0)