daesung9511 commented on code in PR #52224: URL: https://github.com/apache/airflow/pull/52224#discussion_r2174244302
########## providers/postgres/src/airflow/providers/postgres/hooks/postgres.py: ########## @@ -173,6 +178,24 @@ def get_conn(self) -> connection: self.conn = psycopg2.connect(**conn_args) return self.conn + def _get_pandas_df( Review Comment: Hi, I’ve updated the code based on your feedback. Since the method you pointed out is a private one from the parent DbApiHook, I avoided modifying it directly. Instead, I reimplemented get_df here and used the private method only for the Polars case. Let me know if you’d recommend a different approach. I’ll also add unit tests for this soon @bugraoz93 -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: commits-unsubscr...@airflow.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org