From 93b2f6ee3422e35f0fb55461bf6a82079aaa6e12 Mon Sep 17 00:00:00 2001 From: Andrey Velichkevich Date: Tue, 10 Sep 2024 11:53:32 +0100 Subject: [PATCH] [SDK] Read namespace from current context Signed-off-by: Andrey Velichkevich --- sdk/python/kubeflow/training/utils/utils.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/sdk/python/kubeflow/training/utils/utils.py b/sdk/python/kubeflow/training/utils/utils.py index 4f728d9ad3..56011e94aa 100644 --- a/sdk/python/kubeflow/training/utils/utils.py +++ b/sdk/python/kubeflow/training/utils/utils.py @@ -24,6 +24,7 @@ from kubeflow.training import models from kubeflow.training.constants import constants +from kubernetes import config logger = logging.getLogger(__name__) @@ -68,7 +69,11 @@ def is_running_in_k8s(): def get_default_target_namespace(): if not is_running_in_k8s(): - return constants.DEFAULT_NAMESPACE + try: + _, current_context = config.list_kube_config_contexts() + return current_context["context"]["namespace"] + except Exception: + return constants.DEFAULT_NAMESPACE with open("/var/run/secrets/kubernetes.io/serviceaccount/namespace", "r") as f: return f.readline()