Rety upto 3 times while building tensorflow notebook images (#559)

kubeflow · Apr 2, 2018 · 8f8e57d · 8f8e57d
1 parent 0823a40
commit 8f8e57d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/components/tensorflow-notebook-image/releaser/components/workflows.libsonnet b/components/tensorflow-notebook-image/releaser/components/workflows.libsonnet
@@ -82,6 +82,12 @@
       // command: List to pass as the container command.
       local buildTemplate(step_name, command, env_vars=[], sidecars=[]) = {
         name: step_name,
+        // The tensorflow notebook image builds are flaky because they are very
+        // large builds and sometimes there are timeouts while downloading
+        // some pip packages. Retry upto 3 times before giving up.
+        retryStrategy: {
+          limit: 3,
+        },
         container: {
           command: command,
           image: params.step_image,