diff --git a/airflow/dags/DAG_PHT_run_train.py b/airflow/dags/DAG_PHT_run_train.py
index 151dcdd..5a9713b 100644
--- a/airflow/dags/DAG_PHT_run_train.py
+++ b/airflow/dags/DAG_PHT_run_train.py
@@ -334,7 +334,7 @@ def post_run_protocol(train_state):
 
     @task()
     def push_train_image(train_state):
-        client = docker.from_env()
+        client = docker.from_env(timeout=600)
 
         registry_address = os.getenv("HARBOR_URL").split("//")[-1]