DSL - return self for .after() (#435)

kubeflow · Dec 1, 2018 · 9d38f71 · 9d38f71
1 parent 55e39ea
commit 9d38f71
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sdk/python/kfp/dsl/_container_op.py b/sdk/python/kfp/dsl/_container_op.py
@@ -91,6 +91,7 @@ def __init__(self, name: str, image: str, command: str=None, arguments: str=None
   def after(self, op):
     """Specify explicit dependency on another op."""
     self.dependent_op_names.append(op.name)
+    return self
 
   def _validate_memory_string(self, memory_string):
     """Validate a given string is valid for memory request or limit."""