[SDK] Fix Failed condition in wait Job API

Signed-off-by: Andrey Velichkevich <[email protected]>
kubeflow · Jul 9, 2024 · 8909f2b · 8909f2b
1 parent 2b39d3c
commit 8909f2b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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(