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

xiaohanhuang · Mar 4, 2020 · a3897f8 · a3897f8
1 parent acb3d4d
commit a3897f8
Showing 1 changed file with 3 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,18 @@ 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.
     """
-    super(Condition, self).__init__('condition')
+    super(Condition, self).__init__('condition', name)
     self.condition = condition