From d3a340458781b8cff7a52ec55e61bb469c18092b Mon Sep 17 00:00:00 2001 From: helenxie-bit Date: Tue, 8 Oct 2024 22:04:32 -0700 Subject: [PATCH] adjust 'list_namespaced_persistent_volume_claim' to be called with keyword argument Signed-off-by: helenxie-bit --- sdk/python/v1beta1/kubeflow/katib/api/katib_client.py | 2 +- sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py b/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py index c6a4f95c126..707639aa8ba 100644 --- a/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py +++ b/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py @@ -570,7 +570,7 @@ class name in this argument. ) except Exception as e: pvc_list = self.core_api.list_namespaced_persistent_volume_claim( - namespace + namespace=namespace ) # Check if the PVC with the specified name exists. for pvc in pvc_list.items: diff --git a/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py b/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py index 7a833b6dcf3..afe9c8f2ee5 100644 --- a/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py +++ b/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py @@ -65,7 +65,7 @@ def create_namespaced_persistent_volume_claim_response(*args, **kwargs): def list_namespaced_persistent_volume_claim_response(*args, **kwargs): - if args[0] == "pvc creation failed": + if kwargs.get("namespace") == "pvc creation failed": mock_pvc = Mock() mock_pvc.metadata.name = "pvc_failed" mock_list = Mock()