adjust 'list_namespaced_persistent_volume_claim' to be called with ke…

…yword argument Signed-off-by: helenxie-bit <[email protected]>
kubeflow · Oct 9, 2024 · d3a3404 · d3a3404
1 parent e68fe38
commit d3a3404
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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()