Updated error handling

project-codeflare · Aug 21, 2023 · 77a82e2 · 77a82e2
1 parent 16d23d7
commit 77a82e2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/codeflare_sdk/cluster/auth.py b/src/codeflare_sdk/cluster/auth.py
@@ -22,6 +22,7 @@
 import abc
 from kubernetes import client, config
 import os
+from ..utils.kube_api_helpers import _kube_api_error_handling
 
 global api_client
 api_client = None
@@ -168,12 +169,12 @@ def config_check() -> str:
             try:
                 config.load_kube_config()
             except Exception as e:  # pragma: no cover
-                print(e)
+                _kube_api_error_handling(e)
         elif "KUBERNETES_PORT" in os.environ:
             try:
                 config.load_incluster_config()
             except Exception as e:  # pragma: no cover
-                print(e)
+                _kube_api_error_handling(e)
         else:
             raise PermissionError(
                 "Action not permitted, have you put in correct/up-to-date auth credentials?"