MaksYermak commented on code in PR #50341:
URL: https://github.com/apache/airflow/pull/50341#discussion_r2079637519
##########
providers/google/src/airflow/providers/google/cloud/hooks/bigquery.py:
##########
@@ -306,6 +311,26 @@ def get_pandas_df(
return read_gbq(sql, project_id=project_id, dialect=dialect,
credentials=credentials, **kwargs)
+ async def _get_polars_df(self, sql, parameters=None, dialect=None,
**kwargs):
+ try:
+ import polars as pl
+ except ImportError:
+ raise AirflowOptionalProviderFeatureException(
+ "Polars is not installed. Please install it with `pip install
polars`."
+ )
+
+ client = self.get_client()
+ rows = await client.query(sql).result()
+ return pl.from_arrow(rows.to_arrow())
+
+ @airflow_providers_deprecated(
+ reason="Replaced by function `get_df_by_chunks`.",
Review Comment:
@guan404ming for google provider we have a specific deprecation policy
described here
https://airflow.apache.org/docs/apache-airflow-providers-google/stable/deprecation-policy.html.
Could you please update your code using this policy?
--
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
