Added training DAG

interTwin-eu · Jul 7, 2024 · c3cf690 · c3cf690
1 parent 1627995
commit c3cf690
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/dags_airflow_ale_trainmod.py b/dags_airflow_ale_trainmod.py
@@ -26,6 +26,9 @@
     endpoint="train",
     data=json.dumps({"user":"airflow","token":"airflow"}),
     headers={"Content-Type": "application/json"},
+    xcom_push=True,
+    retries=3,
+    retry_delay=timedelta(minutes=5),
     dag=dag,
 )