fix: increace timeout for docker clilent in puch_train_image for larg…

…er images
PHT-Medic · Nov 16, 2023 · 3ea40ac · 3ea40ac
1 parent c62dde9
commit 3ea40ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/airflow/dags/DAG_PHT_run_train.py b/airflow/dags/DAG_PHT_run_train.py
@@ -370,7 +370,7 @@ def post_run_protocol(train_state):
 
     @task()
     def push_train_image(train_state):
-        client = docker.from_env()
+        client = docker.from_env(timeout = 1200)
 
         registry_address = os.getenv("HARBOR_URL").split("//")[-1]