make token var name more consistent

redhat-ai-services · May 14, 2024 · 407c866 · 407c866
1 parent 6a6c5ce
commit 407c866
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 8 deletions.
diff --git a/pipelines/01_test_connection_via_service.py b/pipelines/01_test_connection_via_service.py
@@ -8,7 +8,6 @@
 load_dotenv(override=True)
 
 kubeflow_endpoint = "https://ds-pipeline-dspa:8443"
-bearer_token = os.environ["BEARER_TOKEN"]
 
 if __name__ == "__main__":
     # Check if the script is running in a k8s pod
@@ -17,9 +16,9 @@
     sa_token_path = "/run/secrets/kubernetes.io/serviceaccount/token"  # noqa: S105
     if os.path.isfile(sa_token_path):
         with open(sa_token_path) as f:
-            token = f.read().rstrip()
+            bearer_token = f.read().rstrip()
     else:
-        token = os.environ["BEARER_TOKEN"]
+        bearer_token = os.environ["BEARER_TOKEN"]
 
     # Check if the script is running in a k8s pod
     # Get the CA from the service account if it is

diff --git a/pipelines/02_submitted_pipeline_via_service.py b/pipelines/02_submitted_pipeline_via_service.py
@@ -36,9 +36,9 @@ def add_pipeline(a: float = 1.0, b: float = 7.0):
     sa_token_path = "/run/secrets/kubernetes.io/serviceaccount/token"  # noqa: S105
     if os.path.isfile(sa_token_path):
         with open(sa_token_path) as f:
-            token = f.read().rstrip()
+            bearer_token = f.read().rstrip()
     else:
-        token = os.environ["BEARER_TOKEN"]
+        bearer_token = os.environ["BEARER_TOKEN"]
 
     # Check if the script is running in a k8s pod
     # Get the CA from the service account if it is
@@ -51,7 +51,7 @@ def add_pipeline(a: float = 1.0, b: float = 7.0):
 
     client = kfp.Client(
         host=kubeflow_endpoint,
-        existing_token=token,
+        existing_token=bearer_token,
         ssl_ca_cert=ssl_ca_cert,
     )
 

diff --git a/pipelines/11_iris_training_pipeline.py b/pipelines/11_iris_training_pipeline.py
@@ -207,9 +207,9 @@ def iris_pipeline(model_obc: str = "iris-model"):
     sa_token_path = "/run/secrets/kubernetes.io/serviceaccount/token"  # noqa: S105
     if os.path.isfile(sa_token_path):
         with open(sa_token_path) as f:
-            token = f.read().rstrip()
+            bearer_token = f.read().rstrip()
     else:
-        token = os.environ["BEARER_TOKEN"]
+        bearer_token = os.environ["BEARER_TOKEN"]
 
     # Check if the script is running in a k8s pod
     # Get the CA from the service account if it is