Extend deadline for PyTorch image building (kubeflow#5991)

aws-kf-ci-bot · Jun 18, 2021 · f573b22 · f573b22
1 parent d67b39c
commit f573b22
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/py/kubeflow/kubeflow/cd/notebook_servers/notebook_server_jupyter_pytorch.py b/py/kubeflow/kubeflow/cd/notebook_servers/notebook_server_jupyter_pytorch.py
@@ -15,4 +15,5 @@ def create_workflow(name=None, namespace=None, bucket=None, **kwargs):
                          destination=config.NOTEBOOK_SERVER_JUPYTER_PYTORCH,
                          second_dockerfile="components/example-notebook-servers/jupyter-pytorch/cuda.Dockerfile",
                          second_destination=config.NOTEBOOK_SERVER_JUPYTER_PYTORCH_CUDA,
-                         mem_override="8Gi")
+                         mem_override="8Gi",
+                         deadline_override=6000)
diff --git a/py/kubeflow/kubeflow/cd/notebook_servers/notebook_server_jupyter_pytorch_full.py b/py/kubeflow/kubeflow/cd/notebook_servers/notebook_server_jupyter_pytorch_full.py
@@ -15,4 +15,5 @@ def create_workflow(name=None, namespace=None, bucket=None, **kwargs):
                          destination=config.NOTEBOOK_SERVER_JUPYTER_PYTORCH_FULL,
                          second_dockerfile="components/example-notebook-servers/jupyter-pytorch-full/cuda.Dockerfile",
                          second_destination=config.NOTEBOOK_SERVER_JUPYTER_PYTORCH_CUDA_FULL,
-                         mem_override="8Gi")
+                         mem_override="8Gi",
+                         deadline_override=6000)