This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new fb10108eec add workgroup to templated fields (#31574)
fb10108eec is described below
commit fb10108eec4d9c621f5bcdba3120d9a88a368e7f
Author: simonprydden <[email protected]>
AuthorDate: Sat May 27 20:30:55 2023 +0100
add workgroup to templated fields (#31574)
Co-authored-by: Simon Prydden <[email protected]>
---
airflow/providers/amazon/aws/operators/athena.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/airflow/providers/amazon/aws/operators/athena.py
b/airflow/providers/amazon/aws/operators/athena.py
index 1d54498569..b7c0954f15 100644
--- a/airflow/providers/amazon/aws/operators/athena.py
+++ b/airflow/providers/amazon/aws/operators/athena.py
@@ -40,7 +40,7 @@ class AthenaOperator(BaseOperator):
:param output_location: s3 path to write the query results into.
(templated)
:param aws_conn_id: aws connection to use
:param client_request_token: Unique token created by user to avoid
multiple executions of same query
- :param workgroup: Athena workgroup in which query will be run
+ :param workgroup: Athena workgroup in which query will be run. (templated)
:param query_execution_context: Context in which query need to be run
:param result_configuration: Dict with path to store results in and config
related to encryption
:param sleep_time: Time (in seconds) to wait between two consecutive calls
to check query status on Athena
@@ -51,7 +51,7 @@ class AthenaOperator(BaseOperator):
"""
ui_color = "#44b5e2"
- template_fields: Sequence[str] = ("query", "database", "output_location")
+ template_fields: Sequence[str] = ("query", "database", "output_location",
"workgroup")
template_ext: Sequence[str] = (".sql",)
template_fields_renderers = {"query": "sql"}
