From 199096f4ff18e54ae10dd16099faf358e3fd3643 Mon Sep 17 00:00:00 2001 From: Elad Kalif <45845474+eladkal@users.noreply.github.com> Date: Wed, 16 Apr 2025 11:54:41 +0300 Subject: [PATCH] update deprecated functions to use the stable functions --- .../common/sql/src/airflow/providers/common/sql/hooks/sql.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/providers/common/sql/src/airflow/providers/common/sql/hooks/sql.py b/providers/common/sql/src/airflow/providers/common/sql/hooks/sql.py index 739ed677085e6..aa818f29f4dee 100644 --- a/providers/common/sql/src/airflow/providers/common/sql/hooks/sql.py +++ b/providers/common/sql/src/airflow/providers/common/sql/hooks/sql.py @@ -397,7 +397,7 @@ def get_pandas_df( :param parameters: The parameters to render the SQL query with. :param kwargs: (optional) passed into pandas.io.sql.read_sql method """ - return self._get_pandas_df(sql, parameters, **kwargs) + return self.get_df(sql, parameters, df_type="pandas", **kwargs) @deprecated( reason="Replaced by function `get_df_by_chunks`.", @@ -412,7 +412,7 @@ def get_pandas_df_by_chunks( chunksize: int, **kwargs, ) -> Generator[DataFrame, None, None]: - return self._get_pandas_df_by_chunks(sql, parameters, chunksize=chunksize, **kwargs) + return self.get_df_by_chunks(sql, parameters, chunksize=chunksize, df_type="pandas", **kwargs) def get_df( self,