Adding resiliency to the repo clone command

GoogleCloudPlatform · Mar 22, 2024 · e8140ad · e8140ad
1 parent 006c296
commit e8140ad
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dags/solutions_team/configs/tensorflow/common.py b/dags/solutions_team/configs/tensorflow/common.py
@@ -120,7 +120,8 @@ def set_up_keras(version: Optional[str] = None) -> tuple[str]:
 
   return (
       cmd_install_keras,
-      "export PATH=$PATH:/root/google-cloud-sdk/bin && cd /tmp && sudo gcloud source repos clone tf2-api-tests --project=cloud-ml-auto-solutions",
+      "export PATH=$PATH:/root/google-cloud-sdk/bin && cd /tmp",
+      "gcloud source repos clone tf2-api-tests --project=cloud-ml-auto-solutions || (cd tf2-api-tests && git pull)"
       "cd /tmp/tf2-api-tests && pip install behave matplotlib",
   )