fix env var check

redhat-ai-services · Jul 30, 2024 · ced24c1 · ced24c1
1 parent d24ee22
commit ced24c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pipelines/11_iris_training_pipeline.py b/pipelines/11_iris_training_pipeline.py
@@ -232,7 +232,7 @@ def iris_pipeline(model_obc: str = "iris-model"):
     print(f"Connecting to kfp: {kubeflow_endpoint}")
 
     sa_token_path = "/run/secrets/kubernetes.io/serviceaccount/token"  # noqa: S105
-    if os.environ["BEARER_TOKEN"]:
+    if "BEARER_TOKEN" in os.environ:
         bearer_token = os.environ["BEARER_TOKEN"]
     elif os.path.isfile(sa_token_path):
         with open(sa_token_path) as f: