Add Sentry failure callback to custom python operator

stellar · Aug 19, 2024 · 3d15111 · 3d15111
1 parent 0e4f7b6
commit 3d15111
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dags/stellar_etl_airflow/build_del_ins_operator.py b/dags/stellar_etl_airflow/build_del_ins_operator.py
@@ -1,5 +1,5 @@
 from airflow.operators.python import PythonOperator
-
+from stellar_etl_airflow.default import alert_after_max_retries
 
 def initialize_task_vars(
     table_id,
@@ -70,5 +70,6 @@ def create_del_ins_task(dag, task_vars, del_ins_callable):
         python_callable=del_ins_callable,
         op_kwargs=task_vars,
         provide_context=True,
+        on_failure_callback=alert_after_max_retries,
         dag=dag,
     )