Increase timeout

Signed-off-by: Yuki Iwai <[email protected]>
kubeflow · Nov 8, 2024 · b01ce34 · b01ce34
1 parent 09e73b5
commit b01ce34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/e2e/mpi_job_test.go b/test/e2e/mpi_job_test.go
@@ -554,7 +554,7 @@ func waitForCompletion(ctx context.Context, mpiJob *kubeflow.MPIJob) *kubeflow.M
 	var err error
 
 	ginkgo.By("Waiting for MPIJob to finish")
-	err = wait.PollUntilContextTimeout(ctx, waitInterval, foreverTimeout, false, func(ctx context.Context) (bool, error) {
+	err = wait.PollUntilContextTimeout(ctx, waitInterval, 10*time.Minute, false, func(ctx context.Context) (bool, error) {
 		updatedJob, err := mpiClient.KubeflowV2beta1().MPIJobs(mpiJob.Namespace).Get(ctx, mpiJob.Name, metav1.GetOptions{})
 		if err != nil {
 			return false, err