Skip to content

Commit

Permalink
[SDK] Add more unit tests for TrainingClient APIs - get_job_pods
Browse files Browse the repository at this point in the history
Signed-off-by: yelias <yossi.elias@nokia.com>
  • Loading branch information
yelias committed Jul 18, 2024
1 parent f3ec9e5 commit 6ee09cd
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions sdk/python/kubeflow/training/api/training_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -1012,12 +1012,10 @@ def get_job_pods(
label_selector=label_selector,
async_req=True,
)
print(f"\n\t Here {namespace=}, {label_selector=}\n {thread=}\n")
return thread.get(timeout).items
except multiprocessing.TimeoutError:
raise TimeoutError(f"Timeout to list pods for Job: {namespace}/{name}")
except Exception as e:
print(e)
except Exception:
raise RuntimeError(f"Failed to list pods for Job: {namespace}/{name}")

def get_job_pod_names(
Expand Down

0 comments on commit 6ee09cd

Please sign in to comment.