Skip host name preprocess for the IAP case (kubeflow#3427)

chensun · Apr 3, 2020 · 9b5fcdb · 9b5fcdb
1 parent 25ff7be
commit 9b5fcdb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sdk/python/kfp/_client.py b/sdk/python/kfp/_client.py
@@ -124,7 +124,9 @@ def _load_config(self, host, client_id, namespace, other_client_id, other_client
 
     host = host or ''
     # Preprocess the host endpoint to prevent some common user mistakes.
-    host = re.sub(r'^(http|https)://', '', host).rstrip('/')
+    # This should only be done for non-IAP cases (when client_id is None). IAP requires preserving the protocol.
+    if not client_id:
+      host = re.sub(r'^(http|https)://', '', host).rstrip('/')
 
     if host:
       config.host = host