Skip to content

Commit 2743bab

Browse files
test(sdk): add patch for load_kube_config.
Signed-off-by: Electronic-Waste <[email protected]>
1 parent 13b1d31 commit 2743bab

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

sdk/python/kubeflow/training/api/training_client_test.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -334,7 +334,9 @@ def training_client():
334334

335335
@pytest.fixture
336336
def training_client_wait_for_job_conditions():
337-
with patch.object(TrainingClient, "get_job", side_effect=get_job_response):
337+
with patch.object(TrainingClient, "get_job", side_effect=get_job_response), patch(
338+
"kubernetes.config.load_kube_config", return_value=Mock()
339+
):
338340
client = TrainingClient(job_kind=constants.PYTORCHJOB_KIND)
339341
yield client
340342

0 commit comments

Comments
 (0)