SDK - ContainerOp.set_display_name should return self (#1718)

/assign @gaoning777
2019-08-05 20:47:53 -07:00 · 2019-08-05 20:47:53 -07:00 · 4fb329eaf8
parent 2b72b7f147
commit 4fb329eaf8
1 changed files with 1 additions and 0 deletions
--- a/sdk/python/kfp/dsl/_container_op.py
+++ b/sdk/python/kfp/dsl/_container_op.py
@ -877,6 +877,7 @@ class BaseOp(object):

    def set_display_name(self, name: str):
        self.display_name = name
+        return self

    def __repr__(self):
        return str({self.__class__.__name__: self.__dict__})