apache-airflow-providers-amazon 8.29.0__py3-none-any.whl → 9.0.0__py3-none-any.whl

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

@@ -56,13 +56,16 @@ class RedshiftDataOperator(AwsBaseOperator[RedshiftDataHook]):
     :param workgroup_name: name of the Redshift Serverless workgroup. Mutually exclusive with
         `cluster_identifier`. Specify this parameter to query Redshift Serverless. More info
         https://docs.aws.amazon.com/redshift/latest/mgmt/working-with-serverless.html
+    :param session_id: the session identifier of the query
+    :param session_keep_alive_seconds: duration in seconds to keep the session alive after the query
+        finishes. The maximum time a session can keep alive is 24 hours
     :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:
+    :param verify: Whether 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
@@ -77,6 +80,7 @@ class RedshiftDataOperator(AwsBaseOperator[RedshiftDataHook]):
         "parameters",
         "statement_name",
         "workgroup_name",
+        "session_id",
     )
     template_ext = (".sql",)
     template_fields_renderers = {"sql": "sql"}
@@ -84,8 +88,8 @@ class RedshiftDataOperator(AwsBaseOperator[RedshiftDataHook]):
 
     def __init__(
         self,
-        database: str,
         sql: str | list,
+        database: str | None = None,
         cluster_identifier: str | None = None,
         db_user: str | None = None,
         parameters: list | None = None,
@@ -97,6 +101,8 @@ class RedshiftDataOperator(AwsBaseOperator[RedshiftDataHook]):
         return_sql_result: bool = False,
         workgroup_name: str | None = None,
         deferrable: bool = conf.getboolean("operators", "default_deferrable", fallback=False),
+        session_id: str | None = None,
+        session_keep_alive_seconds: int | None = None,
         **kwargs,
     ) -> None:
         super().__init__(**kwargs)
@@ -120,6 +126,8 @@ class RedshiftDataOperator(AwsBaseOperator[RedshiftDataHook]):
         self.return_sql_result = return_sql_result
         self.statement_id: str | None = None
         self.deferrable = deferrable
+        self.session_id = session_id
+        self.session_keep_alive_seconds = session_keep_alive_seconds
 
     def execute(self, context: Context) -> GetStatementResultResponseTypeDef | str:
         """Execute a statement against Amazon Redshift."""
@@ -130,7 +138,7 @@ class RedshiftDataOperator(AwsBaseOperator[RedshiftDataHook]):
         if self.deferrable:
             wait_for_completion = False
 
-        self.statement_id = self.hook.execute_query(
+        query_execution_output = self.hook.execute_query(
             database=self.database,
             sql=self.sql,
             cluster_identifier=self.cluster_identifier,
@@ -142,8 +150,15 @@ class RedshiftDataOperator(AwsBaseOperator[RedshiftDataHook]):
             with_event=self.with_event,
             wait_for_completion=wait_for_completion,
             poll_interval=self.poll_interval,
+            session_id=self.session_id,
+            session_keep_alive_seconds=self.session_keep_alive_seconds,
         )
 
+        self.statement_id = query_execution_output.statement_id
+
+        if query_execution_output.session_id:
+            self.xcom_push(context, key="session_id", value=query_execution_output.session_id)
+
         if self.deferrable and self.wait_for_completion:
             is_finished = self.hook.check_query_is_finished(self.statement_id)
             if not is_finished:

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

@@ -24,6 +24,9 @@ import sys
 from tempfile import NamedTemporaryFile
 from typing import TYPE_CHECKING, Sequence
 
+import pytz
+from dateutil import parser
+
 from airflow.exceptions import AirflowException
 from airflow.models import BaseOperator
 from airflow.providers.amazon.aws.hooks.s3 import S3Hook
@@ -498,8 +501,8 @@ class S3DeleteObjectsOperator(BaseOperator):
         bucket: str,
         keys: str | list | None = None,
         prefix: str | None = None,
-        from_datetime: datetime | None = None,
-        to_datetime: datetime | None = None,
+        from_datetime: datetime | str | None = None,
+        to_datetime: datetime | str | None = None,
         aws_conn_id: str | None = "aws_default",
         verify: str | bool | None = None,
         **kwargs,
@@ -530,6 +533,13 @@ class S3DeleteObjectsOperator(BaseOperator):
 
         if isinstance(self.keys, (list, str)) and not self.keys:
             return
+        # handle case where dates are strings, specifically when sent as template fields and macros.
+        if isinstance(self.to_datetime, str):
+            self.to_datetime = parser.parse(self.to_datetime).replace(tzinfo=pytz.UTC)
+
+        if isinstance(self.from_datetime, str):
+            self.from_datetime = parser.parse(self.from_datetime).replace(tzinfo=pytz.UTC)
+
         s3_hook = S3Hook(aws_conn_id=self.aws_conn_id, verify=self.verify)
 
         keys = self.keys or s3_hook.list_keys(

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

@@ -19,14 +19,13 @@ from __future__ import annotations
 import datetime
 import json
 import time
-import warnings
 from functools import cached_property
 from typing import TYPE_CHECKING, Any, Callable, Sequence
 
 from botocore.exceptions import ClientError
 
 from airflow.configuration import conf
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.models import BaseOperator
 from airflow.providers.amazon.aws.hooks.base_aws import AwsBaseHook
 from airflow.providers.amazon.aws.hooks.sagemaker import (
@@ -239,7 +238,7 @@ class SageMakerProcessingOperator(SageMakerBaseOperator):
         doesn't finish within max_ingestion_time seconds. If you set this parameter to None,
         the operation does not timeout.
     :param action_if_job_exists: Behaviour if the job name already exists. Possible options are "timestamp"
-        (default), "increment" (deprecated) and "fail".
+        (default) and "fail".
     :param deferrable: Run operator in the deferrable mode. This is only effective if wait_for_completion is
         set to True.
     :return Dict: Returns The ARN of the processing job created in Amazon SageMaker.
@@ -260,18 +259,11 @@ class SageMakerProcessingOperator(SageMakerBaseOperator):
         **kwargs,
     ):
         super().__init__(config=config, aws_conn_id=aws_conn_id, **kwargs)
-        if action_if_job_exists not in ("increment", "fail", "timestamp"):
+        if action_if_job_exists not in ("fail", "timestamp"):
             raise AirflowException(
-                f"Argument action_if_job_exists accepts only 'timestamp', 'increment' and 'fail'. \
+                f"Argument action_if_job_exists accepts only 'timestamp' and 'fail'. \
                 Provided value: '{action_if_job_exists}'."
             )
-        if action_if_job_exists == "increment":
-            warnings.warn(
-                "Action 'increment' on job name conflict has been deprecated for performance reasons."
-                "The alternative to 'fail' is now 'timestamp'.",
-                AirflowProviderDeprecationWarning,
-                stacklevel=2,
-            )
         self.action_if_job_exists = action_if_job_exists
         self.wait_for_completion = wait_for_completion
         self.print_log = print_log
@@ -657,7 +649,7 @@ class SageMakerTransformOperator(SageMakerBaseOperator):
     :param check_if_job_exists: If set to true, then the operator will check whether a transform job
         already exists for the name in the config.
     :param action_if_job_exists: Behaviour if the job name already exists. Possible options are "timestamp"
-        (default), "increment" (deprecated) and "fail".
+        (default) and "fail".
         This is only relevant if check_if_job_exists is True.
     :return Dict: Returns The ARN of the model created in Amazon SageMaker.
     """
@@ -684,18 +676,11 @@ class SageMakerTransformOperator(SageMakerBaseOperator):
         self.max_attempts = max_attempts or 60
         self.max_ingestion_time = max_ingestion_time
         self.check_if_job_exists = check_if_job_exists
-        if action_if_job_exists in ("increment", "fail", "timestamp"):
-            if action_if_job_exists == "increment":
-                warnings.warn(
-                    "Action 'increment' on job name conflict has been deprecated for performance reasons."
-                    "The alternative to 'fail' is now 'timestamp'.",
-                    AirflowProviderDeprecationWarning,
-                    stacklevel=2,
-                )
+        if action_if_job_exists in ("fail", "timestamp"):
             self.action_if_job_exists = action_if_job_exists
         else:
             raise AirflowException(
-                f"Argument action_if_job_exists accepts only 'timestamp', 'increment' and 'fail'. \
+                f"Argument action_if_job_exists accepts only 'timestamp' and 'fail'. \
                 Provided value: '{action_if_job_exists}'."
             )
         self.check_if_model_exists = check_if_model_exists
@@ -1064,7 +1049,7 @@ class SageMakerTrainingOperator(SageMakerBaseOperator):
     :param check_if_job_exists: If set to true, then the operator will check whether a training job
         already exists for the name in the config.
     :param action_if_job_exists: Behaviour if the job name already exists. Possible options are "timestamp"
-        (default), "increment" (deprecated) and "fail".
+        (default) and "fail".
         This is only relevant if check_if_job_exists is True.
     :param deferrable: Run operator in the deferrable mode. This is only effective if wait_for_completion is
         set to True.
@@ -1093,18 +1078,11 @@ class SageMakerTrainingOperator(SageMakerBaseOperator):
         self.max_attempts = max_attempts or 60
         self.max_ingestion_time = max_ingestion_time
         self.check_if_job_exists = check_if_job_exists
-        if action_if_job_exists in {"timestamp", "increment", "fail"}:
-            if action_if_job_exists == "increment":
-                warnings.warn(
-                    "Action 'increment' on job name conflict has been deprecated for performance reasons."
-                    "The alternative to 'fail' is now 'timestamp'.",
-                    AirflowProviderDeprecationWarning,
-                    stacklevel=2,
-                )
+        if action_if_job_exists in {"timestamp", "fail"}:
             self.action_if_job_exists = action_if_job_exists
         else:
             raise AirflowException(
-                f"Argument action_if_job_exists accepts only 'timestamp', 'increment' and 'fail'. \
+                f"Argument action_if_job_exists accepts only 'timestamp' and 'fail'. \
                 Provided value: '{action_if_job_exists}'."
             )
         self.deferrable = deferrable

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

@@ -21,12 +21,9 @@ from __future__ import annotations
 
 import json
 import re
-import warnings
 from functools import cached_property
 from typing import Any
-from urllib.parse import unquote
 
-from airflow.exceptions import AirflowProviderDeprecationWarning
 from airflow.providers.amazon.aws.utils import trim_none_values
 from airflow.secrets import BaseSecretsBackend
 from airflow.utils.log.logging_mixin import LoggingMixin
@@ -145,28 +142,7 @@ class SecretsManagerBackend(BaseSecretsBackend, LoggingMixin):
         self.variables_lookup_pattern = variables_lookup_pattern
         self.config_lookup_pattern = config_lookup_pattern
         self.sep = sep
-
-        if kwargs.pop("full_url_mode", None) is not None:
-            warnings.warn(
-                "The `full_url_mode` kwarg is deprecated. Going forward, the `SecretsManagerBackend`"
-                " will support both URL-encoded and JSON-encoded secrets at the same time. The encoding"
-                " of the secret will be determined automatically.",
-                AirflowProviderDeprecationWarning,
-                stacklevel=2,
-            )
-
-        if kwargs.get("are_secret_values_urlencoded") is not None:
-            warnings.warn(
-                "The `secret_values_are_urlencoded` is deprecated. This kwarg only exists to assist in"
-                " migrating away from URL-encoding secret values for JSON secrets."
-                " To remove this warning, make sure your JSON secrets are *NOT* URL-encoded, and then"
-                " remove this kwarg from backend_kwargs.",
-                AirflowProviderDeprecationWarning,
-                stacklevel=2,
-            )
-            self.are_secret_values_urlencoded = kwargs.pop("are_secret_values_urlencoded", None)
-        else:
-            self.are_secret_values_urlencoded = False
+        self.are_secret_values_urlencoded = False
 
         self.extra_conn_words = extra_conn_words or {}
 
@@ -222,19 +198,6 @@ class SecretsManagerBackend(BaseSecretsBackend, LoggingMixin):
 
         return conn_d
 
-    def _remove_escaping_in_secret_dict(self, secret: dict[str, Any]) -> dict[str, Any]:
-        """Un-escape secret values that are URL-encoded."""
-        for k, v in secret.copy().items():
-            if k == "extra" and isinstance(v, dict):
-                # The old behavior was that extras were _not_ urlencoded inside the secret.
-                # So we should just allow the extra dict to remain as-is.
-                continue
-
-            elif v is not None:
-                secret[k] = unquote(v)
-
-        return secret
-
     def get_conn_value(self, conn_id: str) -> str | None:
         """
         Get serialized representation of Connection.
@@ -259,8 +222,6 @@ class SecretsManagerBackend(BaseSecretsBackend, LoggingMixin):
 
             secret_dict = json.loads(secret)
             standardized_secret_dict = self._standardize_secret_keys(secret_dict)
-            if self.are_secret_values_urlencoded:
-                standardized_secret_dict = self._remove_escaping_in_secret_dict(standardized_secret_dict)
             standardized_secret = json.dumps(standardized_secret_dict)
             return standardized_secret
         else:

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

@@ -20,10 +20,8 @@ from datetime import timedelta
 from functools import cached_property
 from typing import TYPE_CHECKING, Any, Sequence
 
-from deprecated import deprecated
-
 from airflow.configuration import conf
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.providers.amazon.aws.hooks.batch_client import BatchClientHook
 from airflow.providers.amazon.aws.triggers.batch import BatchJobTrigger
 from airflow.sensors.base import BaseSensorOperator
@@ -120,11 +118,6 @@ class BatchSensor(BaseSensorOperator):
         job_id = event["job_id"]
         self.log.info("Batch Job %s complete", job_id)
 
-    @deprecated(reason="use `hook` property instead.", category=AirflowProviderDeprecationWarning)
-    def get_hook(self) -> BatchClientHook:
-        """Create and return a BatchClientHook."""
-        return self.hook
-
     @cached_property
     def hook(self) -> BatchClientHook:
         return BatchClientHook(

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

@@ -19,9 +19,7 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, Iterable, Sequence
 
-from deprecated import deprecated
-
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.providers.amazon.aws.hooks.dms import DmsHook
 from airflow.providers.amazon.aws.sensors.base_aws import AwsBaseSensor
 from airflow.providers.amazon.aws.utils.mixins import aws_template_fields
@@ -68,11 +66,6 @@ class DmsTaskBaseSensor(AwsBaseSensor[DmsHook]):
         self.target_statuses: Iterable[str] = target_statuses or []
         self.termination_statuses: Iterable[str] = termination_statuses or []
 
-    @deprecated(reason="use `hook` property instead.", category=AirflowProviderDeprecationWarning)
-    def get_hook(self) -> DmsHook:
-        """Get DmsHook."""
-        return self.hook
-
     def poke(self, context: Context):
         if not (status := self.hook.get_task_status(self.replication_task_arn)):
             raise AirflowException(

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

@@ -18,6 +18,8 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, Any, Iterable, Sequence
 
+from botocore.exceptions import ClientError
+
 from airflow.providers.amazon.aws.hooks.dynamodb import DynamoDBHook
 from airflow.providers.amazon.aws.sensors.base_aws import AwsBaseSensor
 from airflow.providers.amazon.aws.utils.mixins import aws_template_fields
@@ -102,14 +104,26 @@ class DynamoDBValueSensor(AwsBaseSensor[DynamoDBHook]):
         table = self.hook.conn.Table(self.table_name)
         self.log.info("Table: %s", table)
         self.log.info("Key: %s", key)
-        response = table.get_item(Key=key)
+
         try:
-            item_attribute_value = response["Item"][self.attribute_name]
-            self.log.info("Response: %s", response)
-            self.log.info("Want: %s = %s", self.attribute_name, self.attribute_value)
-            self.log.info("Got: {response['Item'][self.attribute_name]} = %s", item_attribute_value)
-            return item_attribute_value in (
-                [self.attribute_value] if isinstance(self.attribute_value, str) else self.attribute_value
+            response = table.get_item(Key=key)
+        except ClientError as err:
+            self.log.error(
+                "Couldn't get %s from table %s.\nError Code: %s\nError Message: %s",
+                key,
+                self.table_name,
+                err.response["Error"]["Code"],
+                err.response["Error"]["Message"],
             )
-        except KeyError:
             return False
+        else:
+            try:
+                item_attribute_value = response["Item"][self.attribute_name]
+                self.log.info("Response: %s", response)
+                self.log.info("Want: %s = %s", self.attribute_name, self.attribute_value)
+                self.log.info("Got: {response['Item'][self.attribute_name]} = %s", item_attribute_value)
+                return item_attribute_value in (
+                    [self.attribute_value] if isinstance(self.attribute_value, str) else self.attribute_value
+                )
+            except KeyError:
+                return False

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

@@ -21,12 +21,9 @@ from datetime import timedelta
 from functools import cached_property
 from typing import TYPE_CHECKING, Any, Iterable, Sequence
 
-from deprecated import deprecated
-
 from airflow.configuration import conf
 from airflow.exceptions import (
     AirflowException,
-    AirflowProviderDeprecationWarning,
 )
 from airflow.providers.amazon.aws.hooks.emr import EmrContainerHook, EmrHook, EmrServerlessHook
 from airflow.providers.amazon.aws.links.emr import EmrClusterLink, EmrLogsLink, get_log_uri
@@ -68,10 +65,6 @@ class EmrBaseSensor(BaseSensorOperator):
         self.target_states: Iterable[str] = []  # will be set in subclasses
         self.failed_states: Iterable[str] = []  # will be set in subclasses
 
-    @deprecated(reason="use `hook` property instead.", category=AirflowProviderDeprecationWarning)
-    def get_hook(self) -> EmrHook:
-        return self.hook
-
     @cached_property
     def hook(self) -> EmrHook:
         return EmrHook(aws_conn_id=self.aws_conn_id)

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

@@ -20,10 +20,8 @@ from __future__ import annotations
 from datetime import timedelta
 from typing import TYPE_CHECKING, Any, Sequence
 
-from deprecated import deprecated
-
 from airflow.configuration import conf
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.providers.amazon.aws.hooks.glue_catalog import GlueCatalogHook
 from airflow.providers.amazon.aws.sensors.base_aws import AwsBaseSensor
 from airflow.providers.amazon.aws.triggers.glue import GlueCatalogPartitionTrigger
@@ -129,8 +127,3 @@ class GlueCatalogPartitionSensor(AwsBaseSensor[GlueCatalogHook]):
         if event["status"] != "success":
             raise AirflowException(f"Trigger error: event is {event}")
         self.log.info("Partition exists in the Glue Catalog")
-
-    @deprecated(reason="use `hook` property instead.", category=AirflowProviderDeprecationWarning)
-    def get_hook(self) -> GlueCatalogHook:
-        """Get the GlueCatalogHook."""
-        return self.hook

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

@@ -19,9 +19,7 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, Sequence
 
-from deprecated import deprecated
-
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.providers.amazon.aws.hooks.glue_crawler import GlueCrawlerHook
 from airflow.providers.amazon.aws.sensors.base_aws import AwsBaseSensor
 from airflow.providers.amazon.aws.utils.mixins import aws_template_fields
@@ -78,8 +76,3 @@ class GlueCrawlerSensor(AwsBaseSensor[GlueCrawlerHook]):
                 raise AirflowException(f"Status: {crawler_status}")
         else:
             return False
-
-    @deprecated(reason="use `hook` property instead.", category=AirflowProviderDeprecationWarning)
-    def get_hook(self) -> GlueCrawlerHook:
-        """Return a new or pre-existing GlueCrawlerHook."""
-        return self.hook

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

@@ -17,12 +17,9 @@
 # under the License.
 from __future__ import annotations
 
-from functools import cached_property
 from typing import TYPE_CHECKING, Sequence
 
-from deprecated import deprecated
-
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.providers.amazon.aws.hooks.quicksight import QuickSightHook
 from airflow.providers.amazon.aws.sensors.base_aws import AwsBaseSensor
 
@@ -76,28 +73,3 @@ class QuickSightSensor(AwsBaseSensor[QuickSightHook]):
             error = self.hook.get_error_info(None, self.data_set_id, self.ingestion_id)
             raise AirflowException(f"The QuickSight Ingestion failed. Error info: {error}")
         return quicksight_ingestion_state == self.success_status
-
-    @cached_property
-    @deprecated(
-        reason=(
-            "`QuickSightSensor.quicksight_hook` property is deprecated, "
-            "please use `QuickSightSensor.hook` property instead."
-        ),
-        category=AirflowProviderDeprecationWarning,
-    )
-    def quicksight_hook(self):
-        return self.hook
-
-    @cached_property
-    @deprecated(
-        reason=(
-            "`QuickSightSensor.sts_hook` property is deprecated and will be removed in the future. "
-            "This property used for obtain AWS Account ID, "
-            "please consider to use `QuickSightSensor.hook.account_id` instead"
-        ),
-        category=AirflowProviderDeprecationWarning,
-    )
-    def sts_hook(self):
-        from airflow.providers.amazon.aws.hooks.sts import StsHook
-
-        return StsHook(aws_conn_id=self.aws_conn_id)

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

@@ -20,10 +20,8 @@ from datetime import timedelta
 from functools import cached_property
 from typing import TYPE_CHECKING, Any, Sequence
 
-from deprecated import deprecated
-
 from airflow.configuration import conf
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.providers.amazon.aws.hooks.redshift_cluster import RedshiftHook
 from airflow.providers.amazon.aws.triggers.redshift_cluster import RedshiftClusterTrigger
 from airflow.providers.amazon.aws.utils import validate_execute_complete_event
@@ -98,11 +96,6 @@ class RedshiftClusterSensor(BaseSensorOperator):
             self.log.info("%s completed successfully.", self.task_id)
             self.log.info("Cluster Identifier %s is in %s state", self.cluster_identifier, self.target_status)
 
-    @deprecated(reason="use `hook` property instead.", category=AirflowProviderDeprecationWarning)
-    def get_hook(self) -> RedshiftHook:
-        """Create and return a RedshiftHook."""
-        return self.hook
-
     @cached_property
     def hook(self) -> RedshiftHook:
         return RedshiftHook(aws_conn_id=self.aws_conn_id)

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

@@ -25,15 +25,13 @@ from datetime import datetime, timedelta
 from functools import cached_property
 from typing import TYPE_CHECKING, Any, Callable, Sequence, cast
 
-from deprecated import deprecated
-
 from airflow.configuration import conf
 from airflow.providers.amazon.aws.utils import validate_execute_complete_event
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context
 
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.providers.amazon.aws.hooks.s3 import S3Hook
 from airflow.providers.amazon.aws.triggers.s3 import S3KeysUnchangedTrigger, S3KeyTrigger
 from airflow.sensors.base import BaseSensorOperator, poke_mode_only
@@ -221,11 +219,6 @@ class S3KeySensor(BaseSensorOperator):
         elif event["status"] == "error":
             raise AirflowException(event["message"])
 
-    @deprecated(reason="use `hook` property instead.", category=AirflowProviderDeprecationWarning)
-    def get_hook(self) -> S3Hook:
-        """Create and return an S3Hook."""
-        return self.hook
-
     @cached_property
     def hook(self) -> S3Hook:
         return S3Hook(aws_conn_id=self.aws_conn_id, verify=self.verify)

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

@@ -20,9 +20,7 @@ import time
 from functools import cached_property
 from typing import TYPE_CHECKING, Sequence
 
-from deprecated import deprecated
-
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.providers.amazon.aws.hooks.sagemaker import LogState, SageMakerHook
 from airflow.sensors.base import BaseSensorOperator
 
@@ -45,11 +43,6 @@ class SageMakerBaseSensor(BaseSensorOperator):
         self.aws_conn_id = aws_conn_id
         self.resource_type = resource_type  # only used for logs, to say what kind of resource we are sensing
 
-    @deprecated(reason="use `hook` property instead.", category=AirflowProviderDeprecationWarning)
-    def get_hook(self) -> SageMakerHook:
-        """Get SageMakerHook."""
-        return self.hook
-
     @cached_property
     def hook(self) -> SageMakerHook:
         return SageMakerHook(aws_conn_id=self.aws_conn_id)
@@ -238,7 +231,7 @@ class SageMakerTrainingSensor(SageMakerBaseSensor):
         return SageMakerHook.non_terminal_states
 
     def failed_states(self):
-        return SageMakerHook.failed_states
+        return SageMakerHook.training_failed_states
 
     def get_sagemaker_response(self):
         if self.print_log:

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

@@ -22,10 +22,8 @@ from __future__ import annotations
 from datetime import timedelta
 from typing import TYPE_CHECKING, Any, Collection, Sequence
 
-from deprecated import deprecated
-
 from airflow.configuration import conf
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.providers.amazon.aws.hooks.sqs import SqsHook
 from airflow.providers.amazon.aws.sensors.base_aws import AwsBaseSensor
 from airflow.providers.amazon.aws.triggers.sqs import SqsSensorTrigger
@@ -223,8 +221,3 @@ class SqsSensor(AwsBaseSensor[SqsHook]):
             return True
         else:
             return False
-
-    @deprecated(reason="use `hook` property instead.", category=AirflowProviderDeprecationWarning)
-    def get_hook(self) -> SqsHook:
-        """Create and return an SqsHook."""
-        return self.hook

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

@@ -19,9 +19,7 @@ from __future__ import annotations
 import json
 from typing import TYPE_CHECKING, Sequence
 
-from deprecated import deprecated
-
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.providers.amazon.aws.hooks.step_function import StepFunctionHook
 from airflow.providers.amazon.aws.sensors.base_aws import AwsBaseSensor
 from airflow.providers.amazon.aws.utils.mixins import aws_template_fields
@@ -84,8 +82,3 @@ class StepFunctionExecutionSensor(AwsBaseSensor[StepFunctionHook]):
         self.log.info("Doing xcom_push of output")
         self.xcom_push(context, "output", output)
         return True
-
-    @deprecated(reason="use `hook` property instead.", category=AirflowProviderDeprecationWarning)
-    def get_hook(self) -> StepFunctionHook:
-        """Create and return a StepFunctionHook."""
-        return self.hook