diff --git a/providers/src/airflow/providers/amazon/aws/sensors/sqs.py b/providers/src/airflow/providers/amazon/aws/sensors/sqs.py index 006c5bf2ad20b3..016e4fbd6c6d6e 100644 --- a/providers/src/airflow/providers/amazon/aws/sensors/sqs.py +++ b/providers/src/airflow/providers/amazon/aws/sensors/sqs.py @@ -30,10 +30,11 @@ from airflow.providers.amazon.aws.triggers.sqs import SqsSensorTrigger from airflow.providers.amazon.aws.utils import validate_execute_complete_event from airflow.providers.amazon.aws.utils.mixins import aws_template_fields -from airflow.providers.amazon.aws.utils.sqs import MessageFilteringType, process_response +from airflow.providers.amazon.aws.utils.sqs import process_response if TYPE_CHECKING: from airflow.providers.amazon.aws.hooks.base_aws import BaseAwsConnection + from airflow.providers.amazon.aws.utils.sqs import MessageFilteringType from airflow.utils.context import Context diff --git a/providers/src/airflow/providers/amazon/aws/triggers/sqs.py b/providers/src/airflow/providers/amazon/aws/triggers/sqs.py index 31f344b9982090..28c0b509d28c09 100644 --- a/providers/src/airflow/providers/amazon/aws/triggers/sqs.py +++ b/providers/src/airflow/providers/amazon/aws/triggers/sqs.py @@ -22,11 +22,12 @@ from airflow.exceptions import AirflowException from airflow.providers.amazon.aws.hooks.sqs import SqsHook -from airflow.providers.amazon.aws.utils.sqs import MessageFilteringType, process_response +from airflow.providers.amazon.aws.utils.sqs import process_response from airflow.triggers.base import BaseTrigger, TriggerEvent if TYPE_CHECKING: from airflow.providers.amazon.aws.hooks.base_aws import BaseAwsConnection + from airflow.providers.amazon.aws.utils.sqs import MessageFilteringType class SqsSensorTrigger(BaseTrigger):