Skip to content

Commit 2f8889d

Browse files
committed
fixes
1 parent e0440fe commit 2f8889d

File tree

2 files changed

+3
-2
lines changed
  • lib

2 files changed

+3
-2
lines changed

lib/dl_connector_ydb/dl_connector_ydb/core/ydb/adapter.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
from ydb.driver import credentials_impl
1616
import ydb.issues as ydb_cli_err
1717
import ydb_dbapi
18+
import ydb_dbapi.connections
1819

1920
from dl_configs.utils import get_root_certificates
2021
from dl_constants.enums import ConnectionType
@@ -99,7 +100,8 @@ def _list_table_names_i(self, db_name: str, show_dot: bool = False) -> Iterable[
99100
connection = db_engine.connect()
100101
try:
101102
# SA db_engine -> SA connection -> DBAPI connection -> YDB driver
102-
driver = connection.connection._driver # type: ignore # 2024-01-24 # TODO: "DBAPIConnection" has no attribute "_driver" [attr-defined]
103+
dbapi_connection: ydb_dbapi.connections.BaseConnection = connection.connection
104+
driver = dbapi_connection._driver
103105
assert driver
104106

105107
queue = [db_name]

lib/dl_sqlalchemy_ydb/dl_sqlalchemy_ydb/dialect.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,6 @@ class CustomYqlDialect(ydb_sa.YqlDialect):
127127

128128
def __init__(self, *args: typing.Any, **kwargs: typing.Any):
129129
super().__init__(
130-
self,
131130
*args,
132131
**{
133132
**kwargs,

0 commit comments

Comments
 (0)