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, )