SPARK-45959

ahshahid · Nov 24, 2023 · c95b35b · c95b35b
1 parent 10a76ba
commit c95b35b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/execution/python/ExtractPythonUDFsSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/execution/python/ExtractPythonUDFsSuite.scala
@@ -186,7 +186,7 @@ class ExtractPythonUDFsSuite extends SparkPlanTest with SharedSparkSession {
       .withColumn("d", col("c"))
     val pythonEvalNodes4 = collectBatchExec(df4.queryExecution.executedPlan)
     assert(pythonEvalNodes4.size == 1)
-    assert(pythonEvalNodes4.head.udfs.size == 2)
+    assert(pythonEvalNodes4.head.udfs.size == 1)
 
     val df5 = df.withColumns(Seq("c", "d"),
       Seq(batchedNondeterministicPythonUDF(col("a")), batchedNondeterministicPythonUDF(col("a"))))