add optional param (kserve#390)

* add optional param * update * update naming * pylint * fix pylint
SeldonIO · Oct 1, 2019 · 3dcc979 · 3dcc979
1 parent 7c06089
commit 3dcc979
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/python/kfserving/kfserving/api/kf_serving_client.py b/python/kfserving/kfserving/api/kf_serving_client.py
@@ -26,16 +26,17 @@
 class KFServingClient(object):
     '''KFServing Client Apis.'''
 
-    def __init__(self, config_file=None, context=None,
-                 client_configuration=None, persist_config=True):
-        if utils.is_running_in_k8s():
-            config.load_incluster_config()
-        else:
+    def __init__(self, config_file=None, context=None, # pylint: disable=too-many-arguments
+                 client_configuration=None, persist_config=True,
+                 load_kube_config=False):
+        if load_kube_config or not utils.is_running_in_k8s():
             config.load_kube_config(
                 config_file=config_file,
                 context=context,
                 client_configuration=client_configuration,
                 persist_config=persist_config)
+        else:
+            config.load_incluster_config()
 
         self.api_instance = client.CustomObjectsApi()