Bug fix regarding job kind in wait_for_job_conditions python sdk

Signed-off-by: sailesh duddupudi <[email protected]>
kubeflow · Sep 21, 2024 · 4d6ec3a · 4d6ec3a
1 parent da11d11
commit 4d6ec3a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdk/python/kubeflow/training/api/training_client.py b/sdk/python/kubeflow/training/api/training_client.py
@@ -909,7 +909,7 @@ 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,