diff --git a/providers/apache/pinot/src/airflow/providers/apache/pinot/hooks/pinot.py b/providers/apache/pinot/src/airflow/providers/apache/pinot/hooks/pinot.py index f0a11fc37ec79..6394d0e524b1c 100644 --- a/providers/apache/pinot/src/airflow/providers/apache/pinot/hooks/pinot.py +++ b/providers/apache/pinot/src/airflow/providers/apache/pinot/hooks/pinot.py @@ -295,7 +295,7 @@ class PinotDbApiHook(DbApiHook): def get_conn(self) -> Any: """Establish a connection to pinot broker through pinot dbapi.""" - conn = self.get_connection(self.pinot_broker_conn_id) # type: ignore + conn = self.get_connection(self.get_conn_id()) pinot_broker_conn = connect( host=conn.host,