From 18375e8bb9e102bc190272ba90b4ff6b1ef2730c Mon Sep 17 00:00:00 2001 From: Xiaohan Huang Date: Tue, 3 Mar 2020 18:07:52 -0800 Subject: [PATCH] add an optional name for dsl.Condition (kubeflow#3210) --- sdk/python/kfp/dsl/_ops_group.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/sdk/python/kfp/dsl/_ops_group.py b/sdk/python/kfp/dsl/_ops_group.py index 2cb711118a6..0a28e4123e0 100644 --- a/sdk/python/kfp/dsl/_ops_group.py +++ b/sdk/python/kfp/dsl/_ops_group.py @@ -137,18 +137,19 @@ class Condition(OpsGroup): Example usage: ```python - with Condition(param1=='pizza'): + with Condition(param1=='pizza', '[param1 is pizza]'): op1 = ContainerOp(...) op2 = ContainerOp(...) ``` """ - def __init__(self, condition): + def __init__(self, condition, name = None): """Create a new instance of Condition. Args: condition (ConditionOperator): the condition. + name (str): name of the condition """ - super(Condition, self).__init__('condition') + super(Condition, self).__init__('condition', name) self.condition = condition