patch: retry on sensor failure (#124)

goto · Aug 23, 2023 · 2cac6c9 · 2cac6c9
1 parent 93075be
commit 2cac6c9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ext/scheduler/airflow/__lib.py b/ext/scheduler/airflow/__lib.py
@@ -13,7 +13,7 @@
 from airflow.providers.slack.operators.slack import SlackAPIPostOperator
 from airflow.sensors.base import BaseSensorOperator
 from airflow.utils.state import TaskInstanceState
-from airflow.exceptions import AirflowFailException
+from airflow.exceptions import AirflowException
 from croniter import croniter
 from kubernetes.client import models as k8s
 
@@ -256,7 +256,7 @@ def _are_all_job_runs_successful(self, schedule_time_window_start, schedule_time
             self.log.info("job_run api response :: {}".format(api_response))
         except Exception as e:
             self.log.warning("error while fetching job runs :: {}".format(e))
-            raise AirflowFailException(e)
+            raise AirflowException(e)
         for job_run in api_response['jobRuns']:
             if job_run['state'] != 'success':
                 self.log.info("failed for run :: {}".format(job_run))