Skip to content

Commit 41a8a6f

Browse files
committed
Typing fix
1 parent 343dd0e commit 41a8a6f

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

dlt/destinations/impl/databricks/databricks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -264,7 +264,7 @@ def __init__(self, schema: Schema, config: DatabricksClientConfiguration) -> Non
264264
sql_client = DatabricksSqlClient(config.normalize_dataset_name(schema), config.credentials)
265265
super().__init__(schema, config, sql_client)
266266
self.config: DatabricksClientConfiguration = config
267-
self.sql_client: DatabricksSqlClient = sql_client
267+
self.sql_client: DatabricksSqlClient = sql_client # type: ignore[assignment]
268268
self.type_mapper = DatabricksTypeMapper(self.capabilities)
269269

270270
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
@@ -90,7 +90,7 @@ def execute_sql(
9090
@contextmanager
9191
@raise_database_error
9292
def execute_query(self, query: AnyStr, *args: Any, **kwargs: Any) -> Iterator[DBApiCursor]:
93-
curr: DBApiCursor = None
93+
curr: DBApiCursor
9494
# TODO: Inline param support will be dropped in future databricks driver, switch to :named paramstyle
9595
# This will drop support for cluster runtime v13.x
9696
# db_args: Optional[Dict[str, Any]]
@@ -110,7 +110,7 @@ def execute_query(self, query: AnyStr, *args: Any, **kwargs: Any) -> Iterator[DB
110110
# db_args = kwargs or None
111111

112112
db_args = args or kwargs or None
113-
with self._conn.cursor() as curr:
113+
with self._conn.cursor() as curr: # type: ignore[assignment]
114114
curr.execute(query, db_args)
115115
yield DBApiCursorImpl(curr) # type: ignore[abstract]
116116

0 commit comments

Comments
 (0)