update

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Oct 5, 2022 · 804cd22 · 804cd22
1 parent 8fd3877
commit 804cd22
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/flytekit/core/node.py b/flytekit/core/node.py
@@ -49,7 +49,7 @@ def runs_before(self, other: Node):
 
     def __rshift__(self, other: Node):
         self.runs_before(other)
-        return self
+        return other
 
     @property
     def outputs(self):

diff --git a/flytekit/core/promise.py b/flytekit/core/promise.py
@@ -350,7 +350,7 @@ def __hash__(self):
     def __rshift__(self, other: typing.Union[Promise, VoidPromise]):
         if not self.is_ready:
             self.ref.node.runs_before(other.ref.node)
-        return self
+        return other
 
     def with_var(self, new_var: str) -> Promise:
         if self.is_ready:
@@ -563,7 +563,7 @@ def runs_before(self, other: Any):
 
         def __rshift__(self, other: Any):
             # See comment for runs_before
-            return self
+            return other
 
     return Output(*promises)  # type: ignore
 
@@ -679,7 +679,7 @@ def ref(self) -> NodeOutput:
     def __rshift__(self, other: typing.Union[Promise, VoidPromise]):
         if self.ref:
             self.ref.node.runs_before(other.ref.node)
-        return self
+        return other
 
     def with_overrides(self, *args, **kwargs):
         if self.ref: