add workgroup to templated fields (#31574)

Co-authored-by: Simon Prydden <simon.prydden@sykescottages.co.uk>

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"}