[SDK] Minor fix in wait_for_job_conditions with job_kind python train…

…ing API (kubeflow#2265) * Bug fix regarding job kind in wait_for_job_conditions python sdk Signed-off-by: sailesh duddupudi <[email protected]> * Linting update Signed-off-by: sailesh duddupudi <[email protected]> --------- Signed-off-by: sailesh duddupudi <[email protected]>
YosiElias · Sep 23, 2024 · bd82422 · bd82422
1 parent 0f03c5b
commit bd82422
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sdk/python/kubeflow/training/api/training_client.py b/sdk/python/kubeflow/training/api/training_client.py
@@ -903,7 +903,9 @@ def wait_for_job_conditions(
             )
 
             # Get Job conditions.
-            conditions = self.get_job_conditions(job=job, timeout=timeout)
+            conditions = self.get_job_conditions(
+                job=job, timeout=timeout, job_kind=job_kind
+            )
             if len(conditions) > 0:
                 status_logger(
                     name,