Skip to content

Commit

Permalink
fixes linting errors
Browse files Browse the repository at this point in the history
  • Loading branch information
rudolfix committed Nov 6, 2024
1 parent b0f0ea1 commit 4dcc282
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion dlt/destinations/impl/databricks/databricks.py
Original file line number Diff line number Diff line change
Expand Up @@ -224,7 +224,7 @@ def __init__(
)
super().__init__(schema, config, sql_client)
self.config: DatabricksClientConfiguration = config
self.sql_client: DatabricksSqlClient = sql_client # type: ignore[assignment]
self.sql_client: DatabricksSqlClient = sql_client
self.type_mapper = self.capabilities.get_type_mapper()

def create_load_job(
Expand Down
4 changes: 2 additions & 2 deletions dlt/destinations/impl/databricks/sql_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@
class DatabricksCursorImpl(DBApiCursorImpl):
"""Use native data frame support if available"""

native_cursor: DatabricksSqlCursor # type: ignore[assignment]
native_cursor: DatabricksSqlCursor
vector_size: ClassVar[int] = 2048 # vector size is 2048

def iter_arrow(self, chunk_size: int) -> Generator[ArrowTable, None, None]:
Expand Down Expand Up @@ -144,7 +144,7 @@ def execute_query(self, query: AnyStr, *args: Any, **kwargs: Any) -> Iterator[DB
# db_args = kwargs or None

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

Expand Down

0 comments on commit 4dcc282

Please sign in to comment.