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

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

@@ -18,8 +18,12 @@
 from __future__ import annotations
 
 import time
+from dataclasses import dataclass
 from pprint import pformat
 from typing import TYPE_CHECKING, Any, Iterable
+from uuid import UUID
+
+from pendulum import duration
 
 from airflow.providers.amazon.aws.hooks.base_aws import AwsGenericHook
 from airflow.providers.amazon.aws.utils import trim_none_values
@@ -35,6 +39,14 @@ FAILURE_STATES = {FAILED_STATE, ABORTED_STATE}
 RUNNING_STATES = {"PICKED", "STARTED", "SUBMITTED"}
 
 
+@dataclass
+class QueryExecutionOutput:
+    """Describes the output of a query execution."""
+
+    statement_id: str
+    session_id: str | None
+
+
 class RedshiftDataQueryFailedError(ValueError):
     """Raise an error that redshift data query failed."""
 
@@ -65,8 +77,8 @@ class RedshiftDataHook(AwsGenericHook["RedshiftDataAPIServiceClient"]):
 
     def execute_query(
         self,
-        database: str,
         sql: str | list[str],
+        database: str | None = None,
         cluster_identifier: str | None = None,
         db_user: str | None = None,
         parameters: Iterable | None = None,
@@ -76,23 +88,28 @@ class RedshiftDataHook(AwsGenericHook["RedshiftDataAPIServiceClient"]):
         wait_for_completion: bool = True,
         poll_interval: int = 10,
         workgroup_name: str | None = None,
-    ) -> str:
+        session_id: str | None = None,
+        session_keep_alive_seconds: int | None = None,
+    ) -> QueryExecutionOutput:
         """
         Execute a statement against Amazon Redshift.
 
-        :param database: the name of the database
         :param sql: the SQL statement or list of  SQL statement to run
+        :param database: the name of the database
         :param cluster_identifier: unique identifier of a cluster
         :param db_user: the database username
         :param parameters: the parameters for the SQL statement
         :param secret_arn: the name or ARN of the secret that enables db access
         :param statement_name: the name of the SQL statement
-        :param with_event: indicates whether to send an event to EventBridge
-        :param wait_for_completion: indicates whether to wait for a result, if True wait, if False don't wait
+        :param with_event: whether to send an event to EventBridge
+        :param wait_for_completion: whether to wait for a result
         :param poll_interval: how often in seconds to check the query status
         :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
 
         :returns statement_id: str, the UUID of the statement
         """
@@ -105,7 +122,28 @@ class RedshiftDataHook(AwsGenericHook["RedshiftDataAPIServiceClient"]):
             "SecretArn": secret_arn,
             "StatementName": statement_name,
             "WorkgroupName": workgroup_name,
+            "SessionId": session_id,
+            "SessionKeepAliveSeconds": session_keep_alive_seconds,
         }
+
+        if sum(x is not None for x in (cluster_identifier, workgroup_name, session_id)) != 1:
+            raise ValueError(
+                "Exactly one of cluster_identifier, workgroup_name, or session_id must be provided"
+            )
+
+        if session_id is not None:
+            msg = "session_id must be a valid UUID4"
+            try:
+                if UUID(session_id).version != 4:
+                    raise ValueError(msg)
+            except ValueError:
+                raise ValueError(msg)
+
+        if session_keep_alive_seconds is not None and (
+            session_keep_alive_seconds < 0 or duration(seconds=session_keep_alive_seconds).hours > 24
+        ):
+            raise ValueError("Session keep alive duration must be between 0 and 86400 seconds.")
+
         if isinstance(sql, list):
             kwargs["Sqls"] = sql
             resp = self.conn.batch_execute_statement(**trim_none_values(kwargs))
@@ -115,13 +153,10 @@ class RedshiftDataHook(AwsGenericHook["RedshiftDataAPIServiceClient"]):
 
         statement_id = resp["Id"]
 
-        if bool(cluster_identifier) is bool(workgroup_name):
-            raise ValueError("Either 'cluster_identifier' or 'workgroup_name' must be specified.")
-
         if wait_for_completion:
             self.wait_for_results(statement_id, poll_interval=poll_interval)
 
-        return statement_id
+        return QueryExecutionOutput(statement_id=statement_id, session_id=resp.get("SessionId"))
 
     def wait_for_results(self, statement_id: str, poll_interval: int) -> str:
         while True:
@@ -135,9 +170,9 @@ class RedshiftDataHook(AwsGenericHook["RedshiftDataAPIServiceClient"]):
     def check_query_is_finished(self, statement_id: str) -> bool:
         """Check whether query finished, raise exception is failed."""
         resp = self.conn.describe_statement(Id=statement_id)
-        return self.parse_statement_resposne(resp)
+        return self.parse_statement_response(resp)
 
-    def parse_statement_resposne(self, resp: DescribeStatementResponseTypeDef) -> bool:
+    def parse_statement_response(self, resp: DescribeStatementResponseTypeDef) -> bool:
         """Parse the response of describe_statement."""
         status = resp["Status"]
         if status == FINISHED_STATE:
@@ -179,8 +214,10 @@ class RedshiftDataHook(AwsGenericHook["RedshiftDataAPIServiceClient"]):
         :param table: Name of the target table
         :param database: the name of the database
         :param schema: Name of the target schema, public by default
-        :param sql: the SQL statement or list of  SQL statement to run
         :param cluster_identifier: unique identifier of a cluster
+        :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 db_user: the database username
         :param secret_arn: the name or ARN of the secret that enables db access
         :param statement_name: the name of the SQL statement
@@ -212,7 +249,8 @@ class RedshiftDataHook(AwsGenericHook["RedshiftDataAPIServiceClient"]):
             with_event=with_event,
             wait_for_completion=wait_for_completion,
             poll_interval=poll_interval,
-        )
+        ).statement_id
+
         pk_columns = []
         token = ""
         while True:
@@ -251,4 +289,4 @@ class RedshiftDataHook(AwsGenericHook["RedshiftDataAPIServiceClient"]):
         """
         async with self.async_conn as client:
             resp = await client.describe_statement(Id=statement_id)
-            return self.parse_statement_resposne(resp)
+            return self.parse_statement_response(resp)

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

@@ -28,7 +28,6 @@ import os
 import re
 import shutil
 import time
-import warnings
 from contextlib import suppress
 from copy import deepcopy
 from datetime import datetime
@@ -41,8 +40,6 @@ from typing import TYPE_CHECKING, Any, AsyncIterator, Callable
 from urllib.parse import urlsplit
 from uuid import uuid4
 
-from airflow.providers.common.compat.lineage.hook import get_hook_lineage_collector
-
 if TYPE_CHECKING:
     from mypy_boto3_s3.service_resource import Bucket as S3Bucket, Object as S3ResourceObject
 
@@ -51,16 +48,25 @@ if TYPE_CHECKING:
     with suppress(ImportError):
         from aiobotocore.client import AioBaseClient
 
+from importlib.util import find_spec
+
 from asgiref.sync import sync_to_async
 from boto3.s3.transfer import S3Transfer, TransferConfig
 from botocore.exceptions import ClientError
 
-from airflow.exceptions import AirflowException, AirflowNotFoundException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException, AirflowNotFoundException
 from airflow.providers.amazon.aws.exceptions import S3HookUriParseFailure
 from airflow.providers.amazon.aws.hooks.base_aws import AwsBaseHook
 from airflow.providers.amazon.aws.utils.tags import format_tags
 from airflow.utils.helpers import chunks
 
+if find_spec("airflow.assets"):
+    from airflow.lineage.hook import get_hook_lineage_collector
+else:
+    # TODO: import from common.compat directly after common.compat providers with
+    # asset_compat_lineage_collector released
+    from airflow.providers.amazon.aws.utils.asset_compat_lineage_collector import get_hook_lineage_collector
+
 logger = logging.getLogger(__name__)
 
 
@@ -119,15 +125,6 @@ def provide_bucket_name(func: Callable) -> Callable:
 
                 if "bucket_name" in self.service_config:
                     bound_args.arguments["bucket_name"] = self.service_config["bucket_name"]
-                elif self.conn_config and self.conn_config.schema:
-                    warnings.warn(
-                        "s3 conn_type, and the associated schema field, is deprecated. "
-                        "Please use aws conn_type instead, and specify `bucket_name` "
-                        "in `service_config.s3` within `extras`.",
-                        AirflowProviderDeprecationWarning,
-                        stacklevel=2,
-                    )
-                    bound_args.arguments["bucket_name"] = self.conn_config.schema
 
             return func(*bound_args.args, **bound_args.kwargs)
 
@@ -1113,11 +1110,11 @@ class S3Hook(AwsBaseHook):
 
         client = self.get_conn()
         client.upload_file(filename, bucket_name, key, ExtraArgs=extra_args, Config=self.transfer_config)
-        get_hook_lineage_collector().add_input_dataset(
-            context=self, scheme="file", dataset_kwargs={"path": filename}
+        get_hook_lineage_collector().add_input_asset(
+            context=self, scheme="file", asset_kwargs={"path": filename}
         )
-        get_hook_lineage_collector().add_output_dataset(
-            context=self, scheme="s3", dataset_kwargs={"bucket": bucket_name, "key": key}
+        get_hook_lineage_collector().add_output_asset(
+            context=self, scheme="s3", asset_kwargs={"bucket": bucket_name, "key": key}
         )
 
     @unify_bucket_name_and_key
@@ -1260,8 +1257,8 @@ class S3Hook(AwsBaseHook):
             Config=self.transfer_config,
         )
         # No input because file_obj can be anything - handle in calling function if possible
-        get_hook_lineage_collector().add_output_dataset(
-            context=self, scheme="s3", dataset_kwargs={"bucket": bucket_name, "key": key}
+        get_hook_lineage_collector().add_output_asset(
+            context=self, scheme="s3", asset_kwargs={"bucket": bucket_name, "key": key}
         )
 
     def copy_object(
@@ -1318,11 +1315,11 @@ class S3Hook(AwsBaseHook):
         response = self.get_conn().copy_object(
             Bucket=dest_bucket_name, Key=dest_bucket_key, CopySource=copy_source, **kwargs
         )
-        get_hook_lineage_collector().add_input_dataset(
-            context=self, scheme="s3", dataset_kwargs={"bucket": source_bucket_name, "key": source_bucket_key}
+        get_hook_lineage_collector().add_input_asset(
+            context=self, scheme="s3", asset_kwargs={"bucket": source_bucket_name, "key": source_bucket_key}
         )
-        get_hook_lineage_collector().add_output_dataset(
-            context=self, scheme="s3", dataset_kwargs={"bucket": dest_bucket_name, "key": dest_bucket_key}
+        get_hook_lineage_collector().add_output_asset(
+            context=self, scheme="s3", asset_kwargs={"bucket": dest_bucket_name, "key": dest_bucket_key}
         )
         return response
 
@@ -1443,10 +1440,10 @@ class S3Hook(AwsBaseHook):
 
             file_path.parent.mkdir(exist_ok=True, parents=True)
 
-            get_hook_lineage_collector().add_output_dataset(
+            get_hook_lineage_collector().add_output_asset(
                 context=self,
                 scheme="file",
-                dataset_kwargs={"path": file_path if file_path.is_absolute() else file_path.absolute()},
+                asset_kwargs={"path": file_path if file_path.is_absolute() else file_path.absolute()},
             )
             file = open(file_path, "wb")
         else:
@@ -1458,8 +1455,8 @@ class S3Hook(AwsBaseHook):
                 ExtraArgs=self.extra_args,
                 Config=self.transfer_config,
             )
-        get_hook_lineage_collector().add_input_dataset(
-            context=self, scheme="s3", dataset_kwargs={"bucket": bucket_name, "key": key}
+        get_hook_lineage_collector().add_input_asset(
+            context=self, scheme="s3", asset_kwargs={"bucket": bucket_name, "key": key}
         )
         return file.name
 

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

@@ -22,7 +22,6 @@ import re
 import tarfile
 import tempfile
 import time
-import warnings
 from collections import Counter, namedtuple
 from datetime import datetime
 from functools import partial
@@ -31,7 +30,7 @@ from typing import Any, AsyncGenerator, Callable, Generator, cast
 from asgiref.sync import sync_to_async
 from botocore.exceptions import ClientError
 
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.providers.amazon.aws.hooks.base_aws import AwsBaseHook
 from airflow.providers.amazon.aws.hooks.logs import AwsLogsHook
 from airflow.providers.amazon.aws.hooks.s3 import S3Hook
@@ -155,6 +154,7 @@ class SageMakerHook(AwsBaseHook):
     endpoint_non_terminal_states = {"Creating", "Updating", "SystemUpdating", "RollingBack", "Deleting"}
     pipeline_non_terminal_states = {"Executing", "Stopping"}
     failed_states = {"Failed"}
+    training_failed_states = {*failed_states, "Stopped"}
 
     def __init__(self, *args, **kwargs):
         super().__init__(client_type="sagemaker", *args, **kwargs)
@@ -309,7 +309,7 @@ class SageMakerHook(AwsBaseHook):
             self.check_training_status_with_log(
                 config["TrainingJobName"],
                 self.non_terminal_states,
-                self.failed_states,
+                self.training_failed_states,
                 wait_for_completion,
                 check_interval,
                 max_ingestion_time,
@@ -1097,9 +1097,6 @@ class SageMakerHook(AwsBaseHook):
         pipeline_name: str,
         display_name: str = "airflow-triggered-execution",
         pipeline_params: dict | None = None,
-        wait_for_completion: bool = False,
-        check_interval: int | None = None,
-        verbose: bool = True,
     ) -> str:
         """
         Start a new execution for a SageMaker pipeline.
@@ -1114,16 +1111,6 @@ class SageMakerHook(AwsBaseHook):
 
         :return: the ARN of the pipeline execution launched.
         """
-        if wait_for_completion or check_interval is not None:
-            warnings.warn(
-                "parameter `wait_for_completion` and `check_interval` are deprecated, "
-                "remove them and call check_status yourself if you want to wait for completion",
-                AirflowProviderDeprecationWarning,
-                stacklevel=2,
-            )
-        if check_interval is None:
-            check_interval = 30
-
         formatted_params = format_tags(pipeline_params, key_label="Name")
 
         try:
@@ -1136,23 +1123,11 @@ class SageMakerHook(AwsBaseHook):
             self.log.error("Failed to start pipeline execution, error: %s", ce)
             raise
 
-        arn = res["PipelineExecutionArn"]
-        if wait_for_completion:
-            self.check_status(
-                arn,
-                "PipelineExecutionStatus",
-                lambda p: self.describe_pipeline_exec(p, verbose),
-                check_interval,
-                non_terminal_states=self.pipeline_non_terminal_states,
-            )
-        return arn
+        return res["PipelineExecutionArn"]
 
     def stop_pipeline(
         self,
         pipeline_exec_arn: str,
-        wait_for_completion: bool = False,
-        check_interval: int | None = None,
-        verbose: bool = True,
         fail_if_not_running: bool = False,
     ) -> str:
         """
@@ -1171,16 +1146,6 @@ class SageMakerHook(AwsBaseHook):
         :return: Status of the pipeline execution after the operation.
             One of 'Executing'|'Stopping'|'Stopped'|'Failed'|'Succeeded'.
         """
-        if wait_for_completion or check_interval is not None:
-            warnings.warn(
-                "parameter `wait_for_completion` and `check_interval` are deprecated, "
-                "remove them and call check_status yourself if you want to wait for completion",
-                AirflowProviderDeprecationWarning,
-                stacklevel=2,
-            )
-        if check_interval is None:
-            check_interval = 10
-
         for retries in reversed(range(5)):
             try:
                 self.conn.stop_pipeline_execution(PipelineExecutionArn=pipeline_exec_arn)
@@ -1212,15 +1177,6 @@ class SageMakerHook(AwsBaseHook):
 
         res = self.describe_pipeline_exec(pipeline_exec_arn)
 
-        if wait_for_completion and res["PipelineExecutionStatus"] in self.pipeline_non_terminal_states:
-            res = self.check_status(
-                pipeline_exec_arn,
-                "PipelineExecutionStatus",
-                lambda p: self.describe_pipeline_exec(p, verbose),
-                check_interval,
-                non_terminal_states=self.pipeline_non_terminal_states,
-            )
-
         return res["PipelineExecutionStatus"]
 
     def create_model_package_group(self, package_group_name: str, package_group_desc: str = "") -> bool:

airflow/providers/amazon/aws/log/s3_task_handler.py

@@ -62,12 +62,7 @@ class S3TaskHandler(FileTaskHandler, LoggingMixin):
         )
 
     def set_context(self, ti: TaskInstance, *, identifier: str | None = None) -> None:
-        # todo: remove-at-min-airflow-version-2.8
-        #   after Airflow 2.8 can always pass `identifier`
-        if getattr(super(), "supports_task_context_logging", False):
-            super().set_context(ti, identifier=identifier)
-        else:
-            super().set_context(ti)
+        super().set_context(ti, identifier=identifier)
         # Local location and remote location is needed to open and
         # upload local log file to S3 remote storage.
         if TYPE_CHECKING:

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

@@ -17,11 +17,10 @@
 from __future__ import annotations
 
 import time
-import warnings
 from datetime import datetime, timedelta
 from typing import TYPE_CHECKING, cast
 
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.operators.python import ShortCircuitOperator
 from airflow.providers.amazon.aws.hooks.appflow import AppflowHook
 from airflow.providers.amazon.aws.operators.base_aws import AwsBaseOperator
@@ -140,7 +139,6 @@ class AppflowRunOperator(AppflowBaseOperator):
         For more information on how to use this operator, take a look at the guide:
         :ref:`howto/operator:AppflowRunOperator`
 
-    :param source: Obsolete, unnecessary for this operator
     :param flow_name: The flow name
     :param poll_interval: how often in seconds to check the query status
     :param aws_conn_id: The Airflow connection used for AWS credentials.
@@ -155,17 +153,10 @@ class AppflowRunOperator(AppflowBaseOperator):
     def __init__(
         self,
         flow_name: str,
-        source: str | None = None,
         poll_interval: int = 20,
         wait_for_completion: bool = True,
         **kwargs,
     ) -> None:
-        if source is not None:
-            warnings.warn(
-                "The `source` parameter is unused when simply running a flow, please remove it.",
-                AirflowProviderDeprecationWarning,
-                stacklevel=2,
-            )
         super().__init__(
             flow_name=flow_name,
             flow_update=False,

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

@@ -26,13 +26,12 @@ AWS Batch services.
 
 from __future__ import annotations
 
-import warnings
 from datetime import timedelta
 from functools import cached_property
 from typing import TYPE_CHECKING, Any, Sequence
 
 from airflow.configuration import conf
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.models import BaseOperator
 from airflow.models.mappedoperator import MappedOperator
 from airflow.providers.amazon.aws.hooks.batch_client import BatchClientHook
@@ -48,7 +47,6 @@ from airflow.providers.amazon.aws.triggers.batch import (
 )
 from airflow.providers.amazon.aws.utils import trim_none_values, validate_execute_complete_event
 from airflow.providers.amazon.aws.utils.task_log_fetcher import AwsTaskLogFetcher
-from airflow.utils.types import NOTSET
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context
@@ -65,7 +63,6 @@ class BatchOperator(BaseOperator):
     :param job_name: the name for the job that will run on AWS Batch (templated)
     :param job_definition: the job definition name on AWS Batch
     :param job_queue: the queue name on AWS Batch
-    :param overrides: DEPRECATED, use container_overrides instead with the same value.
     :param container_overrides: the `containerOverrides` parameter for boto3 (templated)
     :param ecs_properties_override: the `ecsPropertiesOverride` parameter for boto3 (templated)
     :param eks_properties_override: the `eksPropertiesOverride` parameter for boto3 (templated)
@@ -165,7 +162,6 @@ class BatchOperator(BaseOperator):
         job_name: str,
         job_definition: str,
         job_queue: str,
-        overrides: dict | None = None,  # deprecated
         container_overrides: dict | None = None,
         array_properties: dict | None = None,
         ecs_properties_override: dict | None = None,
@@ -196,21 +192,6 @@ class BatchOperator(BaseOperator):
         self.job_queue = job_queue
 
         self.container_overrides = container_overrides
-        # handle `overrides` deprecation in favor of `container_overrides`
-        if overrides:
-            if container_overrides:
-                # disallow setting both old and new params
-                raise AirflowException(
-                    "'container_overrides' replaces the 'overrides' parameter. "
-                    "You cannot specify both. Please remove assignation to the deprecated 'overrides'."
-                )
-            self.container_overrides = overrides
-            warnings.warn(
-                "Parameter `overrides` is deprecated, Please use `container_overrides` instead.",
-                AirflowProviderDeprecationWarning,
-                stacklevel=2,
-            )
-
         self.ecs_properties_override = ecs_properties_override
         self.eks_properties_override = eks_properties_override
         self.node_overrides = node_overrides
@@ -501,17 +482,8 @@ class BatchCreateComputeEnvironmentOperator(BaseOperator):
         aws_conn_id: str | None = None,
         region_name: str | None = None,
         deferrable: bool = conf.getboolean("operators", "default_deferrable", fallback=False),
-        status_retries=NOTSET,
         **kwargs,
     ):
-        if status_retries is not NOTSET:
-            warnings.warn(
-                "The `status_retries` parameter is unused and should be removed. "
-                "It'll be deleted in a future version.",
-                AirflowProviderDeprecationWarning,
-                stacklevel=2,
-            )
-
         super().__init__(**kwargs)
 
         self.compute_environment_name = compute_environment_name

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

@@ -22,9 +22,7 @@ import logging
 import random
 from typing import TYPE_CHECKING, Any, Sequence
 
-from deprecated.classic import deprecated
-
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning, AirflowTaskTimeout
+from airflow.exceptions import AirflowException, AirflowTaskTimeout
 from airflow.providers.amazon.aws.hooks.datasync import DataSyncHook
 from airflow.providers.amazon.aws.operators.base_aws import AwsBaseOperator
 from airflow.providers.amazon.aws.utils.mixins import aws_template_fields
@@ -199,11 +197,6 @@ class DataSyncOperator(AwsBaseOperator[DataSyncHook]):
     def _hook_parameters(self) -> dict[str, Any]:
         return {**super()._hook_parameters, "wait_interval_seconds": self.wait_interval_seconds}
 
-    @deprecated(reason="use `hook` property instead.", category=AirflowProviderDeprecationWarning)
-    def get_hook(self) -> DataSyncHook:
-        """Create and return DataSyncHook."""
-        return self.hook
-
     def execute(self, context: Context):
         # If task_arn was not specified then try to
         # find 0, 1 or many candidate DataSync Tasks to run

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

@@ -18,13 +18,12 @@
 from __future__ import annotations
 
 import re
-import warnings
 from datetime import timedelta
 from functools import cached_property
 from typing import TYPE_CHECKING, Any, Sequence
 
 from airflow.configuration import conf
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
+from airflow.exceptions import AirflowException
 from airflow.providers.amazon.aws.exceptions import EcsOperatorError, EcsTaskFailToStart
 from airflow.providers.amazon.aws.hooks.base_aws import AwsBaseHook
 from airflow.providers.amazon.aws.hooks.ecs import EcsClusterStates, EcsHook, should_retry_eni
@@ -40,7 +39,6 @@ from airflow.providers.amazon.aws.utils.identifiers import generate_uuid
 from airflow.providers.amazon.aws.utils.mixins import aws_template_fields
 from airflow.providers.amazon.aws.utils.task_log_fetcher import AwsTaskLogFetcher
 from airflow.utils.helpers import prune_dict
-from airflow.utils.types import NOTSET
 
 if TYPE_CHECKING:
     import boto3
@@ -258,19 +256,8 @@ class EcsDeregisterTaskDefinitionOperator(EcsBaseOperator):
         self,
         *,
         task_definition: str,
-        wait_for_completion=NOTSET,
-        waiter_delay=NOTSET,
-        waiter_max_attempts=NOTSET,
         **kwargs,
     ):
-        if any(arg is not NOTSET for arg in [wait_for_completion, waiter_delay, waiter_max_attempts]):
-            warnings.warn(
-                "'wait_for_completion' and waiter related params have no effect and are deprecated, "
-                "please remove them.",
-                AirflowProviderDeprecationWarning,
-                stacklevel=2,
-            )
-
         super().__init__(**kwargs)
         self.task_definition = task_definition
 
@@ -311,19 +298,8 @@ class EcsRegisterTaskDefinitionOperator(EcsBaseOperator):
         family: str,
         container_definitions: list[dict],
         register_task_kwargs: dict | None = None,
-        wait_for_completion=NOTSET,
-        waiter_delay=NOTSET,
-        waiter_max_attempts=NOTSET,
         **kwargs,
     ):
-        if any(arg is not NOTSET for arg in [wait_for_completion, waiter_delay, waiter_max_attempts]):
-            warnings.warn(
-                "'wait_for_completion' and waiter related params have no effect and are deprecated, "
-                "please remove them.",
-                AirflowProviderDeprecationWarning,
-                stacklevel=2,
-            )
-
         super().__init__(**kwargs)
         self.family = family
         self.container_definitions = container_definitions