From 200762d6330883a8246a230afa34898d1a5716d3 Mon Sep 17 00:00:00 2001 From: solovyevt Date: Thu, 26 Sep 2019 17:44:54 +0200 Subject: [PATCH] SDK: fix label check for ContainerOP entities --- sdk/python/kfp/compiler/_default_transformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdk/python/kfp/compiler/_default_transformers.py b/sdk/python/kfp/compiler/_default_transformers.py index 34711be6b58..1072fe49589 100644 --- a/sdk/python/kfp/compiler/_default_transformers.py +++ b/sdk/python/kfp/compiler/_default_transformers.py @@ -17,7 +17,7 @@ def add_pod_env(op: BaseOp) -> BaseOp: """Adds pod environment info to ContainerOp. """ - if isinstance(op, ContainerOp) and op.pod_labels and op.pod_labels['add-pod-env'] == 'true': + if isinstance(op, ContainerOp) and op.pod_labels and 'add-pod-env' in op.pod_labels and op.pod_labels['add-pod-env'] == 'true': from kubernetes import client as k8s_client op.container.add_env_variable( k8s_client.V1EnvVar(