add an optional name for dsl.Condition (#3210)

kubeflow · Mar 4, 2020 · 18375e8 · 18375e8
1 parent acb3d4d
commit 18375e8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git 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