Skip to content

Commit 25e0950

Browse files
Update timeout time for the TestTimeoutGpuProvisioning test
1 parent 1445fa6 commit 25e0950

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/kfto/kfto_training_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ func runKFTOPyTorchJob(t *testing.T, image string, gpu Accelerator, numGpus, num
108108
}
109109

110110
// Make sure the PyTorch job succeeded
111-
test.Eventually(PyTorchJob(test, namespace, tuningJob.Name), TestTimeoutDouble).Should(WithTransform(PyTorchJobConditionSucceeded, Equal(corev1.ConditionTrue)))
111+
test.Eventually(PyTorchJob(test, namespace, tuningJob.Name), TestTimeoutGpuProvisioning).Should(WithTransform(PyTorchJobConditionSucceeded, Equal(corev1.ConditionTrue)))
112112
test.T().Logf("PytorchJob %s/%s ran successfully", tuningJob.Namespace, tuningJob.Name)
113113
}
114114

0 commit comments

Comments
 (0)