fix error

snowflakedb · Nov 20, 2023 · f428a7c · f428a7c
1 parent 09d50b5
commit f428a7c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/main/scala/com/snowflake/snowpark/DataFrame.scala b/src/main/scala/com/snowflake/snowpark/DataFrame.scala
@@ -1981,7 +1981,9 @@ class DataFrame private[snowpark] (
       val dfPrefix = DataFrame.generatePrefix('o')
       val renamedDf =
         this.select(this.output.map(_.name).map(aliasIfNeeded(this, _, dfPrefix, dup.toSet)))
-      renamedDf.joinTableFunction(func, partitionByOrderBy)
+      withPlan {
+        TableFunctionJoin(renamedDf.plan, func, partitionByOrderBy)
+      }
     } else {
       originalResult
     }

diff --git a/src/test/scala/com/snowflake/snowpark_test/TableFunctionSuite.scala b/src/test/scala/com/snowflake/snowpark_test/TableFunctionSuite.scala
@@ -365,4 +365,9 @@ class TableFunctionSuite extends TestData {
       result4,
       Seq(Row(1, 1, "1", 101), Row(1, 2, "2", 101), Row(2, 1, "3", 102), Row(2, 2, "4", 102)))
   }
+
+  test("table function join with duplicated column name") {
+    val df = Seq((1, "1,2"), (2, "3,4")).toDF("idx", "value")
+    df.show()
+  }
 }