From ec4880aae2d2a65516a7bfb0fc7f95a8fa3858ac Mon Sep 17 00:00:00 2001 From: Wei Lee Date: Thu, 25 Jan 2024 18:33:10 +0800 Subject: [PATCH] fix(google): set default use_legacy_sql value to False for BigQueryGetDataOperatorAsync --- astronomer/providers/google/cloud/operators/bigquery.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/astronomer/providers/google/cloud/operators/bigquery.py b/astronomer/providers/google/cloud/operators/bigquery.py index 69bdf34e5..4a6c01446 100644 --- a/astronomer/providers/google/cloud/operators/bigquery.py +++ b/astronomer/providers/google/cloud/operators/bigquery.py @@ -67,7 +67,7 @@ class BigQueryGetDataOperatorAsync(BigQueryGetDataOperator): and set `deferrable` param to `True` instead. """ - def __init__(self, *args: Any, **kwargs: Any) -> None: + def __init__(self, *args: Any, use_legacy_sql: bool = False, **kwargs: Any) -> None: warnings.warn( ( "This class is deprecated." @@ -79,6 +79,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None: ) poll_interval: float = kwargs.pop("poll_interval", 4.0) super().__init__(*args, deferrable=True, **kwargs) + self.use_legacy_sql = use_legacy_sql self.poll_interval = poll_interval