test(sdk): add patch for load_kube_config.

Signed-off-by: Electronic-Waste <[email protected]>
kubeflow · Aug 6, 2024 · f889b9f · f889b9f
1 parent e7be3cb
commit f889b9f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sdk/python/kubeflow/training/api/training_client_test.py b/sdk/python/kubeflow/training/api/training_client_test.py
@@ -334,7 +334,9 @@ def training_client():
 
 @pytest.fixture
 def training_client_wait_for_job_conditions():
-    with patch.object(TrainingClient, "get_job", side_effect=get_job_response):
+    with patch.object(TrainingClient, "get_job", side_effect=get_job_response), patch(
+        "kubernetes.config.load_kube_config", return_value=Mock()
+    ):
         client = TrainingClient(job_kind=constants.PYTORCHJOB_KIND)
         yield client