From 804cd22a7beb721685167f36a0dba29a4c37b13e Mon Sep 17 00:00:00 2001
From: Kevin Su <pingsutw@apache.org>
Date: Wed, 5 Oct 2022 14:54:48 -0700
Subject: [PATCH] update

Signed-off-by: Kevin Su <pingsutw@apache.org>
---
 flytekit/core/node.py    | 2 +-
 flytekit/core/promise.py | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/flytekit/core/node.py b/flytekit/core/node.py
index 37fb51c256..95fdebb4eb 100644
--- 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
index 9d6514bd17..8fcee5e477 100644
--- 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: