Add retry in feedback dag (#183)

closes #106 retry is only missing for feedback dag so adding it
astronomer · Dec 1, 2023 · 542f6f3 · 542f6f3
1 parent 8354071
commit 542f6f3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/airflow/dags/feedback/find_example_runs.py b/airflow/dags/feedback/find_example_runs.py
@@ -15,6 +15,8 @@
 from airflow.decorators import dag, task
 from airflow.operators.empty import EmptyOperator
 
+default_args = {"retries": 3, "retry_delay": 30}
+
 
 def get_firestore_client():
     """
@@ -115,6 +117,7 @@ def process_run(run: dict[str, Any]):
 @dag(
     schedule="@daily",
     start_date=datetime(2023, 1, 1),
+    default_args=default_args,
     catchup=False,
 )
 def find_example_runs():