Skip to content

Commit d4d8d2f

Browse files
authored
Fix mypy type errors in apache-drill provider for sqlalchemy 2 upgrade (apache#56795)
1 parent 6014289 commit d4d8d2f

File tree

1 file changed

+2
-1
lines changed
  • providers/apache/drill/src/airflow/providers/apache/drill/hooks

1 file changed

+2
-1
lines changed

providers/apache/drill/src/airflow/providers/apache/drill/hooks/drill.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626

2727
if TYPE_CHECKING:
2828
from sqlalchemy.engine import Connection
29+
from sqlalchemy.pool import PoolProxiedConnection
2930

3031

3132
class DrillHook(DbApiHook):
@@ -48,7 +49,7 @@ class DrillHook(DbApiHook):
4849
hook_name = "Drill"
4950
supports_autocommit = False
5051

51-
def get_conn(self) -> Connection:
52+
def get_conn(self) -> PoolProxiedConnection:
5253
"""Establish a connection to Drillbit."""
5354
conn_md = self.get_connection(self.get_conn_id())
5455
creds = f"{conn_md.login}:{conn_md.password}@" if conn_md.login else ""

0 commit comments

Comments
 (0)