From c3cf690a1750c5f53a3185332972493af54a139d Mon Sep 17 00:00:00 2001
From: Alessio Romano <alessio.romano329@edu.unito.it>
Date: Sun, 7 Jul 2024 17:02:10 +0200
Subject: [PATCH] Added training DAG

---
 dags_airflow_ale_trainmod.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/dags_airflow_ale_trainmod.py b/dags_airflow_ale_trainmod.py
index 9c96a10..887585f 100644
--- 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,
 )