From f91fecf28d7be0a92ef799cb288adcf7582b20bb Mon Sep 17 00:00:00 2001 From: Wei Lee Date: Wed, 23 Aug 2023 18:27:09 +0800 Subject: [PATCH] docs(providers/amazon): update soft_fail util TODO due to https://github.com/apache/airflow/pull/33424 --- airflow/providers/amazon/aws/sensors/batch.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/airflow/providers/amazon/aws/sensors/batch.py b/airflow/providers/amazon/aws/sensors/batch.py index 29c18af643a60f..e82431b17e0e87 100644 --- a/airflow/providers/amazon/aws/sensors/batch.py +++ b/airflow/providers/amazon/aws/sensors/batch.py @@ -25,6 +25,7 @@ from airflow.configuration import conf from airflow.exceptions import AirflowException from airflow.providers.amazon.aws.hooks.batch_client import BatchClientHook +from airflow.providers.amazon.aws.sensors.utils import raise_failed_or_skiping_exception from airflow.providers.amazon.aws.triggers.batch import BatchJobTrigger from airflow.sensors.base import BaseSensorOperator @@ -117,13 +118,9 @@ def execute_complete(self, context: Context, event: dict[str, Any]) -> None: if event["status"] != "success": message = f"Error while running job: {event}" # TODO: remove this if-else block when min_airflow_version is set to higher than the version that - # raise_failed_or_skiping_exception is introduced in BaseSensorOperator - if getattr(self, "raise_failed_or_skiping_exception"): - self.raise_failed_or_skiping_exception(failed_message=message) - else: - from airflow.providers.amazon.aws.sensors.utils import raise_failed_or_skiping_exception + # changed in https://github.com/apache/airflow/pull/33424 is released - raise_failed_or_skiping_exception(soft_fail=self.soft_fail, failed_message=message) + raise_failed_or_skiping_exception(soft_fail=self.soft_fail, failed_message=message) job_id = event["job_id"] self.log.info("Batch Job %s complete", job_id)