diff --git a/providers/vertica/src/airflow/providers/vertica/hooks/vertica.py b/providers/vertica/src/airflow/providers/vertica/hooks/vertica.py index ad09fbc130d4e..55d59bd57cbfd 100644 --- a/providers/vertica/src/airflow/providers/vertica/hooks/vertica.py +++ b/providers/vertica/src/airflow/providers/vertica/hooks/vertica.py @@ -67,7 +67,7 @@ class VerticaHook(DbApiHook): def get_conn(self) -> connect: """Return vertica connection object.""" - conn = self.get_connection(self.vertica_conn_id) # type: ignore + conn = self.get_connection(self.get_conn_id()) conn_config: dict[str, Any] = { "user": conn.login, "password": conn.password or "", @@ -134,7 +134,7 @@ def get_conn(self) -> connect: conn = connect(**conn_config) return conn - @overload # type: ignore[override] + @overload def run( self, sql: str | Iterable[str],