From 8909f2bfd4680ef7a90672b690b0f4df1d6fdfb9 Mon Sep 17 00:00:00 2001 From: Andrey Velichkevich Date: Tue, 9 Jul 2024 12:32:18 +0100 Subject: [PATCH] [SDK] Fix Failed condition in wait Job API Signed-off-by: Andrey Velichkevich --- sdk/python/kubeflow/training/api/training_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sdk/python/kubeflow/training/api/training_client.py b/sdk/python/kubeflow/training/api/training_client.py index 8750fcbb2d..15548a23b3 100644 --- a/sdk/python/kubeflow/training/api/training_client.py +++ b/sdk/python/kubeflow/training/api/training_client.py @@ -904,9 +904,9 @@ def wait_for_job_conditions( if callback: callback(job) - # Raise an exception if Job is Failed and Failed is not expected condition. + # Raise an exception if Job is Failed and Failed is not the expected condition. if ( - constants.JOB_CONDITION_FAILED not in conditions + constants.JOB_CONDITION_FAILED not in expected_conditions and utils.has_condition(conditions, constants.JOB_CONDITION_FAILED) ): raise RuntimeError(