From 6ee09cd7beb5bfd0d42a526c4e5c5dc5241e8e81 Mon Sep 17 00:00:00 2001 From: yelias Date: Thu, 18 Jul 2024 11:54:16 +0300 Subject: [PATCH] [SDK] Add more unit tests for TrainingClient APIs - get_job_pods Signed-off-by: yelias --- sdk/python/kubeflow/training/api/training_client.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/sdk/python/kubeflow/training/api/training_client.py b/sdk/python/kubeflow/training/api/training_client.py index 782863586b..851c483d80 100644 --- a/sdk/python/kubeflow/training/api/training_client.py +++ b/sdk/python/kubeflow/training/api/training_client.py @@ -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(