guan404ming commented on code in PR #50341:
URL: https://github.com/apache/airflow/pull/50341#discussion_r2079812927
##########
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:
Sure, I have updated. Thanks for your suggestion!
--
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
