From a510e7fa7fb1f6d004f63820b6898d3ba6aeb573 Mon Sep 17 00:00:00 2001 From: kyungjunleeme Date: Sun, 13 Jul 2025 17:21:20 +0900 Subject: [PATCH 1/2] DEL: cleanup type ignore in pinot --- .../pinot/src/airflow/providers/apache/pinot/hooks/pinot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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..29fdbe5d74603 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.pinot_broker_conn_id) pinot_broker_conn = connect( host=conn.host, From 4ef8716e5e4a38f54d30a357c15112df8c8d961b Mon Sep 17 00:00:00 2001 From: Kyungjun Lee Date: Sun, 13 Jul 2025 18:05:02 +0900 Subject: [PATCH 2/2] FIX: get_conn_id Co-authored-by: Jens Scheffler <95105677+jscheffl@users.noreply.github.com> --- .../pinot/src/airflow/providers/apache/pinot/hooks/pinot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 29fdbe5d74603..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) + conn = self.get_connection(self.get_conn_id()) pinot_broker_conn = connect( host=conn.host,