fix: increase timeoutin pull docker images to 15 min becouse of large…

… images
PHT-Medic · Jun 29, 2023 · 7952af0 · 7952af0
1 parent b985955
commit 7952af0
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
@@ -107,7 +107,7 @@ def get_train_image_info():
 
     @task()
     def pull_docker_image(train_state):
-        client = docker.from_env()
+        client = docker.from_env(timeout=900)
 
         registry_address = os.getenv("HARBOR_URL").split("//")[-1]
         print(registry_address)