apache-airflow-providers-amazon 8.19.0rc1__py3-none-any.whl → 8.20.0__py3-none-any.whl

airflow/providers/amazon/aws/operators/emr.py

@@ -576,7 +576,7 @@ class EmrContainerOperator(BaseOperator):
                 stacklevel=2,
             )
             if max_polling_attempts and max_polling_attempts != max_tries:
-                raise Exception("max_polling_attempts must be the same value as max_tries")
+                raise ValueError("max_polling_attempts must be the same value as max_tries")
             else:
                 self.max_polling_attempts = max_tries
 
@@ -1253,27 +1253,77 @@ class EmrServerlessStartJobOperator(BaseOperator):
         op_extra_links = []
 
         if isinstance(self, MappedOperator):
+            operator_class = self.operator_class
             enable_application_ui_links = self.partial_kwargs.get(
                 "enable_application_ui_links"
             ) or self.expand_input.value.get("enable_application_ui_links")
-            job_driver = self.partial_kwargs.get("job_driver") or self.expand_input.value.get("job_driver")
+            job_driver = self.partial_kwargs.get("job_driver", {}) or self.expand_input.value.get(
+                "job_driver", {}
+            )
             configuration_overrides = self.partial_kwargs.get(
                 "configuration_overrides"
             ) or self.expand_input.value.get("configuration_overrides")
 
+            # Configuration overrides can either be a list or a dictionary, depending on whether it's passed in as partial or expand.
+            if isinstance(configuration_overrides, list):
+                if any(
+                    [
+                        operator_class.is_monitoring_in_job_override(
+                            self=operator_class,
+                            config_key="s3MonitoringConfiguration",
+                            job_override=job_override,
+                        )
+                        for job_override in configuration_overrides
+                    ]
+                ):
+                    op_extra_links.extend([EmrServerlessS3LogsLink()])
+                if any(
+                    [
+                        operator_class.is_monitoring_in_job_override(
+                            self=operator_class,
+                            config_key="cloudWatchLoggingConfiguration",
+                            job_override=job_override,
+                        )
+                        for job_override in configuration_overrides
+                    ]
+                ):
+                    op_extra_links.extend([EmrServerlessCloudWatchLogsLink()])
+            else:
+                if operator_class.is_monitoring_in_job_override(
+                    self=operator_class,
+                    config_key="s3MonitoringConfiguration",
+                    job_override=configuration_overrides,
+                ):
+                    op_extra_links.extend([EmrServerlessS3LogsLink()])
+                if operator_class.is_monitoring_in_job_override(
+                    self=operator_class,
+                    config_key="cloudWatchLoggingConfiguration",
+                    job_override=configuration_overrides,
+                ):
+                    op_extra_links.extend([EmrServerlessCloudWatchLogsLink()])
+
         else:
+            operator_class = self
             enable_application_ui_links = self.enable_application_ui_links
             configuration_overrides = self.configuration_overrides
             job_driver = self.job_driver
 
+            if operator_class.is_monitoring_in_job_override(
+                "s3MonitoringConfiguration", configuration_overrides
+            ):
+                op_extra_links.extend([EmrServerlessS3LogsLink()])
+            if operator_class.is_monitoring_in_job_override(
+                "cloudWatchLoggingConfiguration", configuration_overrides
+            ):
+                op_extra_links.extend([EmrServerlessCloudWatchLogsLink()])
+
         if enable_application_ui_links:
             op_extra_links.extend([EmrServerlessDashboardLink()])
-            if "sparkSubmit" in job_driver:
+            if isinstance(job_driver, list):
+                if any("sparkSubmit" in ind_job_driver for ind_job_driver in job_driver):
+                    op_extra_links.extend([EmrServerlessLogsLink()])
+            elif "sparkSubmit" in job_driver:
                 op_extra_links.extend([EmrServerlessLogsLink()])
-        if self.is_monitoring_in_job_override("s3MonitoringConfiguration", configuration_overrides):
-            op_extra_links.extend([EmrServerlessS3LogsLink()])
-        if self.is_monitoring_in_job_override("cloudWatchLoggingConfiguration", configuration_overrides):
-            op_extra_links.extend([EmrServerlessCloudWatchLogsLink()])
 
         return tuple(op_extra_links)
 

airflow/providers/amazon/aws/operators/s3.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """This module contains AWS S3 operators."""
+
 from __future__ import annotations
 
 import subprocess

airflow/providers/amazon/aws/operators/sns.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """Publish message to SNS queue."""
+
 from __future__ import annotations
 
 from typing import TYPE_CHECKING, Sequence

airflow/providers/amazon/aws/operators/sqs.py

@@ -15,6 +15,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """Publish message to SQS queue."""
+
 from __future__ import annotations
 
 from typing import TYPE_CHECKING, Sequence

airflow/providers/amazon/aws/secrets/secrets_manager.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """Objects relating to sourcing secrets from AWS Secrets Manager."""
+
 from __future__ import annotations
 
 import json

airflow/providers/amazon/aws/secrets/systems_manager.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """Objects relating to sourcing connections from AWS SSM Parameter Store."""
+
 from __future__ import annotations
 
 import re

airflow/providers/amazon/aws/sensors/base_aws.py

@@ -26,6 +26,7 @@ from airflow.providers.amazon.aws.utils.mixins import (
     aws_template_fields,
 )
 from airflow.sensors.base import BaseSensorOperator
+from airflow.utils.types import NOTSET, ArgNotSet
 
 
 class AwsBaseSensor(BaseSensorOperator, AwsBaseHookMixin[AwsHookType]):
@@ -84,10 +85,12 @@ class AwsBaseSensor(BaseSensorOperator, AwsBaseHookMixin[AwsHookType]):
         region_name: str | None = None,
         verify: bool | str | None = None,
         botocore_config: dict | None = None,
+        region: str | None | ArgNotSet = NOTSET,  # Required for `.partial` signature check
         **kwargs,
     ):
+        additional_params = {} if region is NOTSET else {"region": region}
         hook_params = AwsHookParams.from_constructor(
-            aws_conn_id, region_name, verify, botocore_config, additional_params=kwargs
+            aws_conn_id, region_name, verify, botocore_config, additional_params=additional_params
         )
         super().__init__(**kwargs)
         self.aws_conn_id = hook_params.aws_conn_id

airflow/providers/amazon/aws/sensors/bedrock.py

@@ -0,0 +1,110 @@
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any, Sequence
+
+from airflow.configuration import conf
+from airflow.providers.amazon.aws.sensors.base_aws import AwsBaseSensor
+from airflow.providers.amazon.aws.triggers.bedrock import BedrockCustomizeModelCompletedTrigger
+from airflow.providers.amazon.aws.utils.mixins import aws_template_fields
+
+if TYPE_CHECKING:
+    from airflow.utils.context import Context
+
+from airflow.exceptions import AirflowException, AirflowSkipException
+from airflow.providers.amazon.aws.hooks.bedrock import BedrockHook
+
+
+class BedrockCustomizeModelCompletedSensor(AwsBaseSensor[BedrockHook]):
+    """
+    Poll the state of the model customization job until it reaches a terminal state; fails if the job fails.
+
+    .. seealso::
+        For more information on how to use this sensor, take a look at the guide:
+        :ref:`howto/sensor:BedrockCustomizeModelCompletedSensor`
+
+
+    :param job_name: The name of the Bedrock model customization job.
+
+    :param deferrable: If True, the sensor will operate in deferrable mode. This mode requires aiobotocore
+        module to be installed.
+        (default: False, but can be overridden in config file by setting default_deferrable to True)
+    :param max_retries: Number of times before returning the current state. (default: 75)
+    :param poke_interval: Polling period in seconds to check for the status of the job. (default: 120)
+    :param aws_conn_id: The Airflow connection used for AWS credentials.
+        If this is ``None`` or empty then the default boto3 behaviour is used. If
+        running Airflow in a distributed manner and aws_conn_id is None or
+        empty, then default boto3 configuration would be used (and must be
+        maintained on each worker node).
+    :param region_name: AWS region_name. If not specified then the default boto3 behaviour is used.
+    :param verify: Whether or not to verify SSL certificates. See:
+        https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html
+    :param botocore_config: Configuration dictionary (key-values) for botocore client. See:
+        https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html
+    """
+
+    INTERMEDIATE_STATES = ("InProgress",)
+    FAILURE_STATES = ("Failed", "Stopping", "Stopped")
+    SUCCESS_STATES = ("Completed",)
+    FAILURE_MESSAGE = "Bedrock model customization job sensor failed."
+
+    aws_hook_class = BedrockHook
+    template_fields: Sequence[str] = aws_template_fields("job_name")
+    ui_color = "#66c3ff"
+
+    def __init__(
+        self,
+        *,
+        job_name: str,
+        max_retries: int = 75,
+        poke_interval: int = 120,
+        deferrable: bool = conf.getboolean("operators", "default_deferrable", fallback=False),
+        **kwargs: Any,
+    ) -> None:
+        super().__init__(**kwargs)
+        self.job_name = job_name
+        self.poke_interval = poke_interval
+        self.max_retries = max_retries
+        self.deferrable = deferrable
+
+    def execute(self, context: Context) -> Any:
+        if self.deferrable:
+            self.defer(
+                trigger=BedrockCustomizeModelCompletedTrigger(
+                    job_name=self.job_name,
+                    waiter_delay=int(self.poke_interval),
+                    waiter_max_attempts=self.max_retries,
+                    aws_conn_id=self.aws_conn_id,
+                ),
+                method_name="poke",
+            )
+        else:
+            super().execute(context=context)
+
+    def poke(self, context: Context) -> bool:
+        state = self.hook.conn.get_model_customization_job(jobIdentifier=self.job_name)["status"]
+        self.log.info("Job '%s' state: %s", self.job_name, state)
+
+        if state in self.FAILURE_STATES:
+            # TODO: remove this if block when min_airflow_version is set to higher than 2.7.1
+            if self.soft_fail:
+                raise AirflowSkipException(self.FAILURE_MESSAGE)
+            raise AirflowException(self.FAILURE_MESSAGE)
+
+        return state not in self.INTERMEDIATE_STATES

airflow/providers/amazon/aws/sensors/cloud_formation.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """This module contains sensors for AWS CloudFormation."""
+
 from __future__ import annotations
 
 from typing import TYPE_CHECKING, Sequence

airflow/providers/amazon/aws/sensors/eks.py

@@ -15,6 +15,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """Tracking the state of Amazon EKS Clusters, Amazon EKS managed node groups, and AWS Fargate profiles."""
+
 from __future__ import annotations
 
 from abc import abstractmethod
@@ -114,12 +115,10 @@ class EksBaseSensor(BaseSensorOperator):
         return state == self.target_state
 
     @abstractmethod
-    def get_state(self) -> ClusterStates | NodegroupStates | FargateProfileStates:
-        ...
+    def get_state(self) -> ClusterStates | NodegroupStates | FargateProfileStates: ...
 
     @abstractmethod
-    def get_terminal_states(self) -> frozenset:
-        ...
+    def get_terminal_states(self) -> frozenset: ...
 
 
 class EksClusterStateSensor(EksBaseSensor):

airflow/providers/amazon/aws/sensors/sqs.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """Reads and then deletes the message from SQS queue."""
+
 from __future__ import annotations
 
 from datetime import timedelta
@@ -72,7 +73,7 @@ class SqsSensor(AwsBaseSensor[SqsHook]):
     :param delete_message_on_reception: Default to `True`, the messages are deleted from the queue
         as soon as being consumed. Otherwise, the messages remain in the queue after consumption and
         should be deleted manually.
-    :param deferrable: If True, the sensor will operate in deferrable more. This mode requires aiobotocore
+    :param deferrable: If True, the sensor will operate in deferrable mode. This mode requires aiobotocore
         module to be installed.
         (default: False, but can be overridden in config file by setting default_deferrable to True)
     :param aws_conn_id: The Airflow connection used for AWS credentials.

airflow/providers/amazon/aws/transfers/azure_blob_to_s3.py

@@ -120,8 +120,10 @@ class AzureBlobStorageToS3Operator(BaseOperator):
         )
 
         self.log.info(
-            f"Getting list of the files in Container: {self.container_name}; "
-            f"Prefix: {self.prefix}; Delimiter: {self.delimiter};"
+            "Getting list of the files in Container: %r; Prefix: %r; Delimiter: %r.",
+            self.container_name,
+            self.prefix,
+            self.delimiter,
         )
 
         files = wasb_hook.get_blobs_list_recursive(

airflow/providers/amazon/aws/transfers/base.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """This module contains base AWS to AWS transfer operator."""
+
 from __future__ import annotations
 
 import warnings

airflow/providers/amazon/aws/transfers/exasol_to_s3.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """Transfers data from Exasol database into a S3 Bucket."""
+
 from __future__ import annotations
 
 from tempfile import NamedTemporaryFile

airflow/providers/amazon/aws/transfers/gcs_to_s3.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """This module contains Google Cloud Storage to S3 operator."""
+
 from __future__ import annotations
 
 import os

airflow/providers/amazon/aws/transfers/google_api_to_s3.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """This module allows you to transfer data from any Google API endpoint into a S3 Bucket."""
+
 from __future__ import annotations
 
 import json

airflow/providers/amazon/aws/transfers/hive_to_dynamodb.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """This module contains operator to move data from Hive to DynamoDB."""
+
 from __future__ import annotations
 
 import json

airflow/providers/amazon/aws/transfers/http_to_s3.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """This module contains operator to move data from HTTP endpoint to S3."""
+
 from __future__ import annotations
 
 from functools import cached_property

airflow/providers/amazon/aws/transfers/imap_attachment_to_s3.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """This module allows you to transfer mail attachments from a mail server into s3 bucket."""
+
 from __future__ import annotations
 
 from typing import TYPE_CHECKING, Sequence

airflow/providers/amazon/aws/transfers/redshift_to_s3.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """Transfers data from AWS Redshift into a S3 Bucket."""
+
 from __future__ import annotations
 
 import re
@@ -109,35 +110,19 @@ class RedshiftToS3Operator(BaseOperator):
     ) -> None:
         super().__init__(**kwargs)
         self.s3_bucket = s3_bucket
-        self.s3_key = f"{s3_key}/{table}_" if (table and table_as_file_name) else s3_key
+        self.s3_key = s3_key
         self.schema = schema
         self.table = table
         self.redshift_conn_id = redshift_conn_id
         self.aws_conn_id = aws_conn_id
         self.verify = verify
-        self.unload_options: list = unload_options or []
+        self.unload_options = unload_options or []
         self.autocommit = autocommit
         self.include_header = include_header
         self.parameters = parameters
         self.table_as_file_name = table_as_file_name
         self.redshift_data_api_kwargs = redshift_data_api_kwargs or {}
-
-        if select_query:
-            self.select_query = select_query
-        elif self.schema and self.table:
-            self.select_query = f"SELECT * FROM {self.schema}.{self.table}"
-        else:
-            raise ValueError(
-                "Please provide both `schema` and `table` params or `select_query` to fetch the data."
-            )
-
-        if self.include_header and "HEADER" not in [uo.upper().strip() for uo in self.unload_options]:
-            self.unload_options = [*self.unload_options, "HEADER"]
-
-        if self.redshift_data_api_kwargs:
-            for arg in ["sql", "parameters"]:
-                if arg in self.redshift_data_api_kwargs:
-                    raise AirflowException(f"Cannot include param '{arg}' in Redshift Data API kwargs")
+        self.select_query = select_query
 
     def _build_unload_query(
         self, credentials_block: str, select_query: str, s3_key: str, unload_options: str
@@ -153,9 +138,26 @@ class RedshiftToS3Operator(BaseOperator):
         """
 
     def execute(self, context: Context) -> None:
+        if self.table and self.table_as_file_name:
+            self.s3_key = f"{self.s3_key}/{self.table}_"
+
+        if self.schema and self.table:
+            self.select_query = f"SELECT * FROM {self.schema}.{self.table}"
+
+        if self.select_query is None:
+            raise ValueError(
+                "Please provide both `schema` and `table` params or `select_query` to fetch the data."
+            )
+
+        if self.include_header and "HEADER" not in [uo.upper().strip() for uo in self.unload_options]:
+            self.unload_options = [*self.unload_options, "HEADER"]
+
         redshift_hook: RedshiftDataHook | RedshiftSQLHook
         if self.redshift_data_api_kwargs:
             redshift_hook = RedshiftDataHook(aws_conn_id=self.redshift_conn_id)
+            for arg in ["sql", "parameters"]:
+                if arg in self.redshift_data_api_kwargs:
+                    raise AirflowException(f"Cannot include param '{arg}' in Redshift Data API kwargs")
         else:
             redshift_hook = RedshiftSQLHook(redshift_conn_id=self.redshift_conn_id)
         conn = S3Hook.get_connection(conn_id=self.aws_conn_id) if self.aws_conn_id else None

airflow/providers/amazon/aws/triggers/bedrock.py

@@ -0,0 +1,61 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from airflow.providers.amazon.aws.hooks.bedrock import BedrockHook
+from airflow.providers.amazon.aws.triggers.base import AwsBaseWaiterTrigger
+
+if TYPE_CHECKING:
+    from airflow.providers.amazon.aws.hooks.base_aws import AwsGenericHook
+
+
+class BedrockCustomizeModelCompletedTrigger(AwsBaseWaiterTrigger):
+    """
+    Trigger when a Bedrock model customization job is complete.
+
+    :param job_name: The name of the Bedrock model customization job.
+    :param waiter_delay: The amount of time in seconds to wait between attempts. (default: 120)
+    :param waiter_max_attempts: The maximum number of attempts to be made. (default: 75)
+    :param aws_conn_id: The Airflow connection used for AWS credentials.
+    """
+
+    def __init__(
+        self,
+        *,
+        job_name: str,
+        waiter_delay: int = 120,
+        waiter_max_attempts: int = 75,
+        aws_conn_id: str | None = None,
+    ) -> None:
+        super().__init__(
+            serialized_fields={"job_name": job_name},
+            waiter_name="model_customization_job_complete",
+            waiter_args={"jobIdentifier": job_name},
+            failure_message="Bedrock model customization failed.",
+            status_message="Status of Bedrock model customization job is",
+            status_queries=["status"],
+            return_key="job_name",
+            return_value=job_name,
+            waiter_delay=waiter_delay,
+            waiter_max_attempts=waiter_max_attempts,
+            aws_conn_id=aws_conn_id,
+        )
+
+    def hook(self) -> AwsGenericHook:
+        return BedrockHook(aws_conn_id=self.aws_conn_id)

airflow/providers/amazon/aws/triggers/eks.py

@@ -214,7 +214,7 @@ class EksDeleteClusterTrigger(AwsBaseWaiterTrigger):
                 )
             self.log.info("All Fargate profiles deleted")
         else:
-            self.log.info(f"No Fargate profiles associated with cluster {self.cluster_name}")
+            self.log.info("No Fargate profiles associated with cluster %s", self.cluster_name)
 
 
 class EksCreateFargateProfileTrigger(AwsBaseWaiterTrigger):

airflow/providers/amazon/aws/triggers/redshift_cluster.py

@@ -311,6 +311,7 @@ class RedshiftClusterTrigger(BaseTrigger):
                     "status"
                 ] == "error":
                     yield TriggerEvent(res)
+                    return
                 await asyncio.sleep(self.poke_interval)
         except Exception as e:
             yield TriggerEvent({"status": "error", "message": str(e)})

airflow/providers/amazon/aws/triggers/s3.py

@@ -98,8 +98,9 @@ class S3KeyTrigger(BaseTrigger):
                             )
                             await asyncio.sleep(self.poke_interval)
                             yield TriggerEvent({"status": "running", "files": s3_objects})
-
-                        yield TriggerEvent({"status": "success"})
+                        else:
+                            yield TriggerEvent({"status": "success"})
+                        return
 
                     self.log.info("Sleeping for %s seconds", self.poke_interval)
                     await asyncio.sleep(self.poke_interval)
@@ -204,6 +205,7 @@ class S3KeysUnchangedTrigger(BaseTrigger):
                     )
                     if result.get("status") in ("success", "error"):
                         yield TriggerEvent(result)
+                        return
                     elif result.get("status") == "pending":
                         self.previous_objects = result.get("previous_objects", set())
                         self.last_activity_time = result.get("last_activity_time")

airflow/providers/amazon/aws/triggers/sagemaker.py

@@ -245,8 +245,8 @@ class SageMakerTrainingPrintLogTrigger(BaseTrigger):
         job_already_completed = status not in self.hook.non_terminal_states
         state = LogState.COMPLETE if job_already_completed else LogState.TAILING
         last_describe_job_call = time.time()
-        while True:
-            try:
+        try:
+            while True:
                 (
                     state,
                     last_description,
@@ -267,6 +267,7 @@ class SageMakerTrainingPrintLogTrigger(BaseTrigger):
                     reason = last_description.get("FailureReason", "(No reason provided)")
                     error_message = f"SageMaker job failed because {reason}"
                     yield TriggerEvent({"status": "error", "message": error_message})
+                    return
                 else:
                     billable_seconds = SageMakerHook.count_billable_seconds(
                         training_start_time=last_description["TrainingStartTime"],
@@ -275,5 +276,6 @@ class SageMakerTrainingPrintLogTrigger(BaseTrigger):
                     )
                     self.log.info("Billable seconds: %d", billable_seconds)
                     yield TriggerEvent({"status": "success", "message": last_description})
-            except Exception as e:
-                yield TriggerEvent({"status": "error", "message": str(e)})
+                    return
+        except Exception as e:
+            yield TriggerEvent({"status": "error", "message": str(e)})

airflow/providers/amazon/aws/utils/emailer.py

@@ -16,6 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """Airflow module for email backend using AWS SES."""
+
 from __future__ import annotations
 
 from typing import Any

airflow/providers/amazon/aws/waiters/bedrock.json

@@ -0,0 +1,42 @@
+{
+    "version": 2,
+    "waiters": {
+        "model_customization_job_complete": {
+            "delay": 120,
+            "maxAttempts": 75,
+            "operation": "GetModelCustomizationJob",
+            "acceptors": [
+                {
+                    "matcher": "path",
+                    "argument": "status",
+                    "expected": "InProgress",
+                    "state": "retry"
+                },
+                {
+                    "matcher": "path",
+                    "argument": "status",
+                    "expected": "Completed",
+                    "state": "success"
+                },
+                {
+                    "matcher": "path",
+                    "argument": "status",
+                    "expected": "Failed",
+                    "state": "failure"
+                },
+                {
+                    "matcher": "path",
+                    "argument": "status",
+                    "expected": "Stopping",
+                    "state": "failure"
+                },
+                {
+                    "matcher": "path",
+                    "argument": "status",
+                    "expected": "Stopped",
+                    "state": "failure"
+                }
+            ]
+        }
+    }
+}