fix(google): set default use_legacy_sql value to False for BigQueryGe…

…tDataOperatorAsync
astronomer · Jan 25, 2024 · ec4880a · ec4880a
1 parent 46ee507
commit ec4880a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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