fix

apache · Apr 6, 2023 · 573a262 · 573a262
1 parent 4cef204
commit 573a262
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...c/test/scala/org/apache/kyuubi/engine/spark/operation/SparkArrowbasedOperationSuite.scala b/...c/test/scala/org/apache/kyuubi/engine/spark/operation/SparkArrowbasedOperationSuite.scala
@@ -166,7 +166,8 @@ class SparkArrowbasedOperationSuite extends WithSparkSQLEngine with SparkDataTyp
       val df = spark.sql(s"select * from t_1 limit $size")
       val headPlan = df.queryExecution.executedPlan.collectLeaves().head
       assert(headPlan.isInstanceOf[AdaptiveSparkPlanExec])
-      val finalPhysicalPlan = SparkDatasetHelper.finalPhysicalPlan(AdaptiveSparkPlanExec)
+      val finalPhysicalPlan =
+        SparkDatasetHelper.finalPhysicalPlan(headPlan.asInstanceOf[AdaptiveSparkPlanExec])
       assert(finalPhysicalPlan.isInstanceOf[CollectLimitExec])
 
       val arrowBinary = SparkDatasetHelper.executeArrowBatchCollect(df.queryExecution.executedPlan)