Fix.

ueshin · Sep 27, 2023 · a1cbd19 · a1cbd19
1 parent ca44182
commit a1cbd19
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/pyspark/sql/connect/plan.py b/python/pyspark/sql/connect/plan.py
@@ -933,14 +933,14 @@ def plan(self, session: "SparkConnectClient") -> proto.Relation:
             plan.as_of_join.left_as_of.CopyFrom(self.left_as_of.to_plan(session))
         else:
             plan.as_of_join.left_as_of.CopyFrom(
-                ColumnReference(self.left_as_of, self.left._plan_id)
+                ColumnReference(self.left_as_of, self.left._plan_id).to_plan(session)
             )
 
         if isinstance(self.right_as_of, Column):
             plan.as_of_join.right_as_of.CopyFrom(self.right_as_of.to_plan(session))
         else:
             plan.as_of_join.right_as_of.CopyFrom(
-                ColumnReference(self.right_as_of, self.right._plan_id)
+                ColumnReference(self.right_as_of, self.right._plan_id).to_plan(session)
             )
 
         if self.on is not None: