apache-airflow-providers-amazon 9.6.0__tar.gz → 9.6.1__tar.gz

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: apache-airflow-providers-amazon
-Version: 9.6.0
+Version: 9.6.1
 Summary: Provider package apache-airflow-providers-amazon for Apache Airflow
 Keywords: airflow-provider,amazon,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -36,7 +36,8 @@ Requires-Dist: jmespath>=0.7.0
 Requires-Dist: python3-saml>=1.16.0
 Requires-Dist: xmlsec!=1.3.15,>=1.3.14
 Requires-Dist: sagemaker-studio>=1.0.9
-Requires-Dist: aiobotocore[boto3]>=2.20.0 ; extra == "aiobotocore"
+Requires-Dist: aiobotocore[boto3]>=2.21.1 ; extra == "aiobotocore"
+Requires-Dist: boto3<1.37.2,>=1.37.0 ; extra == "aiobotocore"
 Requires-Dist: apache-airflow-providers-apache-hive ; extra == "apache-hive"
 Requires-Dist: apache-airflow-providers-cncf-kubernetes>=7.2.0 ; extra == "cncf-kubernetes"
 Requires-Dist: apache-airflow-providers-exasol ; extra == "exasol"
@@ -54,8 +55,8 @@ Requires-Dist: apache-airflow-providers-salesforce ; extra == "salesforce"
 Requires-Dist: apache-airflow-providers-ssh ; extra == "ssh"
 Requires-Dist: apache-airflow-providers-standard ; extra == "standard"
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.0/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.0
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.1/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.1
 Project-URL: Mastodon, https://fosstodon.org/@airflow
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
 Project-URL: Source Code, https://github.com/apache/airflow
@@ -103,7 +104,7 @@ Provides-Extra: standard
 
 Package ``apache-airflow-providers-amazon``
 
-Release: ``9.6.0``
+Release: ``9.6.1``
 
 
 Amazon integration (including `Amazon Web Services (AWS) <https://aws.amazon.com/>`__).
@@ -116,7 +117,7 @@ This is a provider package for ``amazon`` provider. All classes for this provide
 are in ``airflow.providers.amazon`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.0/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.1/>`_.
 
 Installation
 ------------
@@ -184,5 +185,5 @@ Dependent package
 ======================================================================================================================  ===================
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.0/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.1/changelog.html>`_.
 

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/README.rst

@@ -23,7 +23,7 @@
 
 Package ``apache-airflow-providers-amazon``
 
-Release: ``9.6.0``
+Release: ``9.6.1``
 
 
 Amazon integration (including `Amazon Web Services (AWS) <https://aws.amazon.com/>`__).
@@ -36,7 +36,7 @@ This is a provider package for ``amazon`` provider. All classes for this provide
 are in ``airflow.providers.amazon`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.0/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.1/>`_.
 
 Installation
 ------------
@@ -104,4 +104,4 @@ Dependent package
 ======================================================================================================================  ===================
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.0/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.1/changelog.html>`_.

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/docs/changelog.rst

@@ -26,6 +26,31 @@
 Changelog
 ---------
 
+9.6.1
+.....
+
+Bug Fixes
+~~~~~~~~~
+
+* ``Fix 'EksClusterStateSensor'. Save 'region' as attribute (#49138)``
+* ``Decrease default value of 'waiter_max_attempts' in 'MwaaTriggerDagRunOperator' (#49136)``
+* ``Increase default value of 'waiter_max_attempts' in 'BedrockBatchInferenceOperator' (#49090)``
+
+Misc
+~~~~
+
+* ``Use contextlib.suppress(exception) instead of try-except-pass and add SIM105 ruff rule (#49251)``
+* ``Add base_url fallback for aws auth_manager (#49305)``
+* ``remove superfluous else block (#49199)``
+* ``AWS Batch Operators/Sensors inherit AWS Base classes (#49172)``
+* ``Help pip to find appropriate boto for aiobotocore (#49166)``
+* ``Update EKS Operators and Sensors to inherit AWS base classes (#48192)``
+
+.. Below changes are excluded from the changelog. Move them to
+   appropriate section above if needed. Do not delete the lines(!):
+   * ``Unpause DAG in AWS MWAA system test (#49145)``
+   * ``Fix AWS auth manager system test (#49072)``
+
 9.6.0
 .....
 

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/docs/commits.rst

@@ -34,6 +34,27 @@ For high-level changelog, see :doc:`package information including changelog <ind
 
 
 
+9.6.1
+.....
+
+Latest change: 2025-04-15
+
+==================================================================================================  ===========  ===================================================================================================
+Commit                                                                                              Committed    Subject
+==================================================================================================  ===========  ===================================================================================================
+`47bd8961a9 <https://github.com/apache/airflow/commit/47bd8961a9ce4e2cea0dbabd400d2508eb291948>`__  2025-04-15   ``Use contextlib.suppress(exception) instead of try-except-pass and add SIM105 ruff rule (#49251)``
+`fe52138b3d <https://github.com/apache/airflow/commit/fe52138b3dcc4aa72312b8ca9160cb9a06a11623>`__  2025-04-15   ``Add base_url fallback for aws auth_manager (#49305)``
+`2a0009dddf <https://github.com/apache/airflow/commit/2a0009dddfdaf26fa9b5746257202a97b5552da0>`__  2025-04-14   ``Unpause DAG in AWS MWAA system test (#49145)``
+`20f8af546e <https://github.com/apache/airflow/commit/20f8af546e63c6178df19166acbda8f768c21bf0>`__  2025-04-14   ``Fix 'EksClusterStateSensor'. Save 'region' as attribute (#49138)``
+`cb295c351a <https://github.com/apache/airflow/commit/cb295c351a016c0a10cab07f2a628b865cff3ca3>`__  2025-04-14   ``remove superfluous else block (#49199)``
+`1781dc9d5f <https://github.com/apache/airflow/commit/1781dc9d5fcbb93bce7018a4050ddd36ec1a65d5>`__  2025-04-13   ``AWS Batch Operators/Sensors inherit AWS Base classes (#49172)``
+`4fa0eec9d4 <https://github.com/apache/airflow/commit/4fa0eec9d472c76be564422c9d0339246d1b5ace>`__  2025-04-12   ``Help pip to find appropriate boto for aiobotocore (#49166)``
+`a4a6e2bc99 <https://github.com/apache/airflow/commit/a4a6e2bc99cbcf95f13af6732ee272961a761aba>`__  2025-04-11   ``Decrease default value of 'waiter_max_attempts' in 'MwaaTriggerDagRunOperator' (#49136)``
+`3d6460e034 <https://github.com/apache/airflow/commit/3d6460e034ad8b4a5e878c54f1959c237f24bc8e>`__  2025-04-11   ``Update EKS Operators and Sensors to inherit AWS base classes (#48192)``
+`28a96413bb <https://github.com/apache/airflow/commit/28a96413bb7705f3626ff56bf4dffc4d3d109846>`__  2025-04-10   ``Increase default value of 'waiter_max_attempts' in 'BedrockBatchInferenceOperator' (#49090)``
+`87306b32c1 <https://github.com/apache/airflow/commit/87306b32c18b42dd5dbe24227076fb4ba76778e0>`__  2025-04-10   ``Fix AWS auth manager system test (#49072)``
+==================================================================================================  ===========  ===================================================================================================
+
 9.6.0
 .....
 
@@ -42,6 +63,7 @@ Latest change: 2025-04-10
 ==================================================================================================  ===========  ============================================================================================
 Commit                                                                                              Committed    Subject
 ==================================================================================================  ===========  ============================================================================================
+`4a8567b20b <https://github.com/apache/airflow/commit/4a8567b20bdd6555cbdc936d6674bf4fa390b0d5>`__  2025-04-10   ``Prepare docs for Apr 2nd wave of providers (#49051)``
 `b58f027cbf <https://github.com/apache/airflow/commit/b58f027cbf39e5eea982a5560b22024e2de328a8>`__  2025-04-10   ``Fix default base value (#49013)``
 `e0eaa604f4 <https://github.com/apache/airflow/commit/e0eaa604f49a612432451910e30e74eaf2563601>`__  2025-04-10   ``Rename list_jobs method to describe_jobs in GlueJobHook (#48904)``
 `f1a12f13e4 <https://github.com/apache/airflow/commit/f1a12f13e486f4c3b9e89fee92d5fccc488073dc>`__  2025-04-09   ``Refactor AWS system tests to not use @task.branch (#48973)``

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/docs/index.rst

@@ -86,7 +86,7 @@ apache-airflow-providers-amazon package
 Amazon integration (including `Amazon Web Services (AWS) <https://aws.amazon.com/>`__).
 
 
-Release: 9.6.0
+Release: 9.6.1
 
 Provider package
 ----------------
@@ -165,5 +165,5 @@ Downloading official packages
 You can download officially released packages and verify their checksums and signatures from the
 `Official Apache Download site <https://downloads.apache.org/airflow/providers/>`_
 
-* `The apache-airflow-providers-amazon 9.6.0 sdist package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_amazon-9.6.0.tar.gz>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_amazon-9.6.0.tar.gz.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_amazon-9.6.0.tar.gz.sha512>`__)
-* `The apache-airflow-providers-amazon 9.6.0 wheel package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_amazon-9.6.0-py3-none-any.whl>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_amazon-9.6.0-py3-none-any.whl.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_amazon-9.6.0-py3-none-any.whl.sha512>`__)
+* `The apache-airflow-providers-amazon 9.6.1 sdist package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_amazon-9.6.1.tar.gz>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_amazon-9.6.1.tar.gz.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_amazon-9.6.1.tar.gz.sha512>`__)
+* `The apache-airflow-providers-amazon 9.6.1 wheel package <https://downloads.apache.org/airflow/providers/apache_airflow_providers_amazon-9.6.1-py3-none-any.whl>`_ (`asc <https://downloads.apache.org/airflow/providers/apache_airflow_providers_amazon-9.6.1-py3-none-any.whl.asc>`__, `sha512 <https://downloads.apache.org/airflow/providers/apache_airflow_providers_amazon-9.6.1-py3-none-any.whl.sha512>`__)

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/docs/operators/batch.rst

@@ -29,6 +29,11 @@ Prerequisite Tasks
 
 .. include:: ../_partials/prerequisite_tasks.rst
 
+Generic Parameters
+------------------
+
+.. include:: ../_partials/generic_parameters.rst
+
 Operators
 ---------
 

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/docs/operators/eks.rst

@@ -31,6 +31,11 @@ Prerequisite Tasks
 
 .. include:: ../_partials/prerequisite_tasks.rst
 
+Generic Parameters
+------------------
+
+.. include:: ../_partials/generic_parameters.rst
+
 Operators
 ---------
 

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/provider.yaml

@@ -22,9 +22,10 @@ description: |
   Amazon integration (including `Amazon Web Services (AWS) <https://aws.amazon.com/>`__).
 
 state: ready
-source-date-epoch: 1744280881
+source-date-epoch: 1744788746
 # note that those versions are maintained by release manager - do not update them manually
 versions:
+  - 9.6.1
   - 9.6.0
   - 9.5.0
   - 9.4.0

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/pyproject.toml

@@ -25,7 +25,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "apache-airflow-providers-amazon"
-version = "9.6.0"
+version = "9.6.1"
 description = "Provider package apache-airflow-providers-amazon for Apache Airflow"
 readme = "README.rst"
 authors = [
@@ -95,7 +95,10 @@ dependencies = [
 # TODO: We can remove it once boto3 and aiobotocore both have compatible botocore version or
 # boto3 have native async support and we move away from aio aiobotocore
 "aiobotocore" = [
-    "aiobotocore[boto3]>=2.20.0",
+    "aiobotocore[boto3]>=2.21.1",
+    # boto3 here should be synchronized with latest aiobotocore version otherwise pip might get
+    # into a backtracking loop and fail to install the package
+    "boto3<1.37.2,>=1.37.0"
 ]
 "cncf.kubernetes" = [
     "apache-airflow-providers-cncf-kubernetes>=7.2.0",
@@ -163,7 +166,7 @@ dev = [
     "apache-airflow-providers-salesforce",
     "apache-airflow-providers-ssh",
     # Additional devel dependencies (do not remove this line and add extra development dependencies)
-    "aiobotocore>=2.13.0",
+    "aiobotocore[boto3]>=2.21.1",
     "apache-airflow-providers-postgres",
     "aws_xray_sdk>=2.12.0",
     "moto[cloudformation,glue]>=5.1.2",
@@ -204,8 +207,8 @@ apache-airflow-providers-common-sql = {workspace = true}
 apache-airflow-providers-standard = {workspace = true}
 
 [project.urls]
-"Documentation" = "https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.0"
-"Changelog" = "https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.0/changelog.html"
+"Documentation" = "https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.1"
+"Changelog" = "https://airflow.apache.org/docs/apache-airflow-providers-amazon/9.6.1/changelog.html"
 "Bug Tracker" = "https://github.com/apache/airflow/issues"
 "Source Code" = "https://github.com/apache/airflow"
 "Slack Chat" = "https://s.apache.org/airflow-slack"

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/__init__.py

@@ -29,7 +29,7 @@ from airflow import __version__ as airflow_version
 
 __all__ = ["__version__"]
 
-__version__ = "9.6.0"
+__version__ = "9.6.1"
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
     "2.9.0"

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/auth_manager/cli/avp_commands.py

@@ -107,26 +107,23 @@ def _create_policy_store(client: BaseClient, args) -> tuple[str | None, bool]:
             f"There is already a policy store with description '{args.policy_store_description}' in Amazon Verified Permissions: '{existing_policy_stores[0]['policyStoreId']}'."
         )
         return existing_policy_stores[0]["policyStoreId"], False
-    else:
-        print(f"No policy store with description '{args.policy_store_description}' found, creating one.")
-        if args.dry_run:
-            print(
-                f"Dry run, not creating the policy store with description '{args.policy_store_description}'."
-            )
-            return None, True
-
-        response = client.create_policy_store(
-            validationSettings={
-                "mode": "STRICT",
-            },
-            description=args.policy_store_description,
-        )
-        if args.verbose:
-            log.debug("Response from create_policy_store: %s", response)
+    print(f"No policy store with description '{args.policy_store_description}' found, creating one.")
+    if args.dry_run:
+        print(f"Dry run, not creating the policy store with description '{args.policy_store_description}'.")
+        return None, True
+
+    response = client.create_policy_store(
+        validationSettings={
+            "mode": "STRICT",
+        },
+        description=args.policy_store_description,
+    )
+    if args.verbose:
+        log.debug("Response from create_policy_store: %s", response)
 
-        print(f"Policy store created: '{response['policyStoreId']}'")
+    print(f"Policy store created: '{response['policyStoreId']}'")
 
-        return response["policyStoreId"], True
+    return response["policyStoreId"], True
 
 
 def _set_schema(client: BaseClient, policy_store_id: str, args) -> None:

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/auth_manager/router/login.py

@@ -91,7 +91,7 @@ def login_callback(request: Request):
 
 def _init_saml_auth(request: Request) -> OneLogin_Saml2_Auth:
     request_data = _prepare_request(request)
-    base_url = conf.get(section="api", key="base_url")
+    base_url = conf.get(section="api", key="base_url", fallback="/")
     settings = {
         # We want to keep this flag on in case of errors.
         # It provides an error reasons, if turned off, it does not

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/executors/ecs/ecs_executor.py

@@ -23,6 +23,7 @@ Each Airflow task gets delegated out to an Amazon ECS Task.
 
 from __future__ import annotations
 
+import contextlib
 import time
 from collections import defaultdict, deque
 from collections.abc import Sequence
@@ -449,13 +450,11 @@ class AwsEcsExecutor(BaseExecutor):
             else:
                 task = run_task_response["tasks"][0]
                 self.active_workers.add_task(task, task_key, queue, cmd, exec_config, attempt_number)
-                try:
-                    self.running_state(task_key, task.task_arn)
-                except AttributeError:
+                with contextlib.suppress(AttributeError):
                     # running_state is newly added, and only needed to support task adoption (an optional
                     # executor feature).
                     # TODO: remove when min airflow version >= 2.9.2
-                    pass
+                    self.running_state(task_key, task.task_arn)
 
     def _run_task(
         self, task_id: TaskInstanceKey, cmd: CommandType, queue: str, exec_config: ExecutorConfigType

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/executors/ecs/ecs_executor_config.py

@@ -72,7 +72,7 @@ def build_task_kwargs() -> dict:
         raise ValueError(
             "capacity_provider_strategy and launch_type are mutually exclusive, you can not provide both."
         )
-    elif "cluster" in task_kwargs and not (has_capacity_provider or has_launch_type):
+    if "cluster" in task_kwargs and not (has_capacity_provider or has_launch_type):
         # Default API behavior if neither is provided is to fall back on the default capacity
         # provider if it exists. Since it is not a required value, check if there is one
         # before using it, and if there is not then use the FARGATE launch_type as

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/executors/ecs/utils.py

@@ -138,7 +138,7 @@ class EcsExecutorTask:
         """
         if self.last_status == "RUNNING":
             return State.RUNNING
-        elif self.desired_status == "RUNNING":
+        if self.desired_status == "RUNNING":
             return State.QUEUED
         is_finished = self.desired_status == "STOPPED"
         has_exit_codes = all(["exit_code" in x for x in self.containers])

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/hooks/athena.py

@@ -208,7 +208,7 @@ class AthenaHook(AwsBaseHook):
         if query_state is None:
             self.log.error("Invalid Query state. Query execution id: %s", query_execution_id)
             return None
-        elif query_state in self.INTERMEDIATE_STATES or query_state in self.FAILURE_STATES:
+        if query_state in self.INTERMEDIATE_STATES or query_state in self.FAILURE_STATES:
             self.log.error(
                 'Query is in "%s" state. Cannot fetch results. Query execution id: %s',
                 query_state,

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/hooks/base_aws.py

@@ -191,15 +191,13 @@ class BaseSessionFactory(LoggingMixin):
                 session = self.get_async_session()
                 self._apply_session_kwargs(session)
                 return session
-            else:
-                return boto3.session.Session(region_name=self.region_name)
-        elif not self.role_arn:
+            return boto3.session.Session(region_name=self.region_name)
+        if not self.role_arn:
             if deferrable:
                 session = self.get_async_session()
                 self._apply_session_kwargs(session)
                 return session
-            else:
-                return self.basic_session
+            return self.basic_session
 
         # Values stored in ``AwsConnectionWrapper.session_kwargs`` are intended to be used only
         # to create the initial boto3 session.
@@ -624,7 +622,7 @@ class AwsGenericHook(BaseHook, Generic[BaseAwsConnection]):
                 if is_resource_type:
                     raise LookupError("Requested `resource_type`, but `client_type` was set instead.")
                 return self.client_type
-            elif self.resource_type:
+            if self.resource_type:
                 if not is_resource_type:
                     raise LookupError("Requested `client_type`, but `resource_type` was set instead.")
                 return self.resource_type
@@ -840,15 +838,14 @@ class AwsGenericHook(BaseHook, Generic[BaseAwsConnection]):
         """
         if "/" in role:
             return role
-        else:
-            session = self.get_session(region_name=region_name)
-            _client = session.client(
-                service_name="iam",
-                endpoint_url=self.conn_config.get_service_endpoint_url("iam"),
-                config=self.config,
-                verify=self.verify,
-            )
-            return _client.get_role(RoleName=role)["Role"]["Arn"]
+        session = self.get_session(region_name=region_name)
+        _client = session.client(
+            service_name="iam",
+            endpoint_url=self.conn_config.get_service_endpoint_url("iam"),
+            config=self.config,
+            verify=self.verify,
+        )
+        return _client.get_role(RoleName=role)["Role"]["Arn"]
 
     @staticmethod
     def retry(should_retry: Callable[[Exception], bool]):

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/hooks/batch_client.py

@@ -142,6 +142,17 @@ class BatchProtocol(Protocol):
         """
         ...
 
+    def create_compute_environment(self, **kwargs) -> dict:
+        """
+        Create an AWS Batch compute environment.
+
+        :param kwargs: Arguments for boto3 create_compute_environment
+
+        .. seealso::
+            - https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/batch/client/create_compute_environment.html
+        """
+        ...
+
 
 # Note that the use of invalid-name parameters should be restricted to the boto3 mappings only;
 # all the Airflow wrappers of boto3 clients should not adopt invalid-names to match boto3.

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/hooks/cloud_formation.py

@@ -62,8 +62,7 @@ class CloudFormationHook(AwsBaseHook):
         except ClientError as e:
             if "does not exist" in str(e):
                 return None
-            else:
-                raise e
+            raise e
 
     def create_stack(self, stack_name: str, cloudformation_parameters: dict) -> None:
         """

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/hooks/datasync.py

@@ -312,9 +312,9 @@ class DataSyncHook(AwsBaseHook):
             self.log.info("status=%s", status)
             if status in self.TASK_EXECUTION_SUCCESS_STATES:
                 return True
-            elif status in self.TASK_EXECUTION_FAILURE_STATES:
+            if status in self.TASK_EXECUTION_FAILURE_STATES:
                 return False
-            elif status is None or status in self.TASK_EXECUTION_INTERMEDIATE_STATES:
+            if status is None or status in self.TASK_EXECUTION_INTERMEDIATE_STATES:
                 time.sleep(self.wait_interval_seconds)
             else:
                 raise AirflowException(f"Unknown status: {status}")  # Should never happen

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/hooks/dms.py

@@ -108,9 +108,8 @@ class DmsHook(AwsBaseHook):
             status = replication_tasks[0]["Status"]
             self.log.info('Replication task with ARN(%s) has status "%s".', replication_task_arn, status)
             return status
-        else:
-            self.log.info("Replication task with ARN(%s) is not found.", replication_task_arn)
-            return None
+        self.log.info("Replication task with ARN(%s) is not found.", replication_task_arn)
+        return None
 
     def create_replication_task(
         self,

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/hooks/dynamodb.py

@@ -102,5 +102,4 @@ class DynamoDBHook(AwsBaseHook):
             error_code = e.response.get("Error", {}).get("Code")
             if error_code == "ImportNotFoundException":
                 raise AirflowException("S3 import into Dynamodb job not found.")
-            else:
-                raise e
+            raise e

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/hooks/emr.py

@@ -83,11 +83,10 @@ class EmrHook(AwsBaseHook):
             cluster_id = matching_clusters[0]["Id"]
             self.log.info("Found cluster name = %s id = %s", emr_cluster_name, cluster_id)
             return cluster_id
-        elif len(matching_clusters) > 1:
+        if len(matching_clusters) > 1:
             raise AirflowException(f"More than one cluster found for name {emr_cluster_name}")
-        else:
-            self.log.info("No cluster found for name %s", emr_cluster_name)
-            return None
+        self.log.info("No cluster found for name %s", emr_cluster_name)
+        return None
 
     def create_job_flow(self, job_flow_overrides: dict[str, Any]) -> dict[str, Any]:
         """
@@ -387,12 +386,11 @@ class EmrContainerHook(AwsBaseHook):
 
         if response["ResponseMetadata"]["HTTPStatusCode"] != 200:
             raise AirflowException(f"Create EMR EKS Cluster failed: {response}")
-        else:
-            self.log.info(
-                "Create EMR EKS Cluster success - virtual cluster id %s",
-                response["id"],
-            )
-            return response["id"]
+        self.log.info(
+            "Create EMR EKS Cluster success - virtual cluster id %s",
+            response["id"],
+        )
+        return response["id"]
 
     def submit_job(
         self,
@@ -446,13 +444,12 @@ class EmrContainerHook(AwsBaseHook):
 
         if response["ResponseMetadata"]["HTTPStatusCode"] != 200:
             raise AirflowException(f"Start Job Run failed: {response}")
-        else:
-            self.log.info(
-                "Start Job Run success - Job Id %s and virtual cluster id %s",
-                response["id"],
-                response["virtualClusterId"],
-            )
-            return response["id"]
+        self.log.info(
+            "Start Job Run success - Job Id %s and virtual cluster id %s",
+            response["id"],
+            response["virtualClusterId"],
+        )
+        return response["id"]
 
     def get_job_failure_reason(self, job_id: str) -> str | None:
         """

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/hooks/glue.py

@@ -320,8 +320,7 @@ class GlueJobHook(AwsBaseHook):
             if ret:
                 time.sleep(sleep_before_return)
                 return ret
-            else:
-                time.sleep(self.job_poll_interval)
+            time.sleep(self.job_poll_interval)
 
     async def async_job_completion(self, job_name: str, run_id: str, verbose: bool = False) -> dict[str, str]:
         """
@@ -338,8 +337,7 @@ class GlueJobHook(AwsBaseHook):
             ret = self._handle_state(job_run_state, job_name, run_id, verbose, next_log_tokens)
             if ret:
                 return ret
-            else:
-                await asyncio.sleep(self.job_poll_interval)
+            await asyncio.sleep(self.job_poll_interval)
 
     def _handle_state(
         self,
@@ -367,13 +365,12 @@ class GlueJobHook(AwsBaseHook):
             job_error_message = f"Exiting Job {run_id} Run State: {state}"
             self.log.info(job_error_message)
             raise AirflowException(job_error_message)
-        else:
-            self.log.info(
-                "Polling for AWS Glue Job %s current run state with status %s",
-                job_name,
-                state,
-            )
-            return None
+        self.log.info(
+            "Polling for AWS Glue Job %s current run state with status %s",
+            job_name,
+            state,
+        )
+        return None
 
     def has_job(self, job_name) -> bool:
         """
@@ -414,8 +411,7 @@ class GlueJobHook(AwsBaseHook):
             self.conn.update_job(JobName=job_name, JobUpdate=job_kwargs)
             self.log.info("Updated configurations: %s", update_config)
             return True
-        else:
-            return False
+        return False
 
     def get_or_create_glue_job(self) -> str | None:
         """

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/hooks/mwaa.py

@@ -101,13 +101,12 @@ class MwaaHook(AwsBaseHook):
                     "Access Denied due to missing airflow:InvokeRestApi in IAM policy. Trying again by generating local token..."
                 )
                 return self._invoke_rest_api_using_local_session_token(**api_kwargs)
-            else:
-                to_log = e.response
-                # ResponseMetadata is removed because it contains data that is either very unlikely to be
-                # useful in XComs and logs, or redundant given the data already included in the response
-                to_log.pop("ResponseMetadata", None)
-                self.log.error(to_log)
-                raise
+            to_log = e.response
+            # ResponseMetadata is removed because it contains data that is either very unlikely to be
+            # useful in XComs and logs, or redundant given the data already included in the response
+            to_log.pop("ResponseMetadata", None)
+            self.log.error(to_log)
+            raise
 
     def _invoke_rest_api_using_local_session_token(
         self,

{apache_airflow_providers_amazon-9.6.0 → apache_airflow_providers_amazon-9.6.1}/src/airflow/providers/amazon/aws/hooks/redshift_data.py

@@ -181,7 +181,7 @@ class RedshiftDataHook(AwsGenericHook["RedshiftDataAPIServiceClient"]):
             if num_rows is not None:
                 self.log.info("Processed %s rows", num_rows)
             return True
-        elif status in FAILURE_STATES:
+        if status in FAILURE_STATES:
             exception_cls = (
                 RedshiftDataQueryFailedError if status == FAILED_STATE else RedshiftDataQueryAbortedError
             )