taken

apache · Dec 6, 2023 · 5358762 · 5358762
1 parent 819e35f
commit 5358762
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...k-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/operation/SparkOperation.scala b/...k-sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/operation/SparkOperation.scala
@@ -248,9 +248,9 @@ abstract class SparkOperation(session: Session)
               RowSet.emptyTRowSet()
             }
           } else {
-            val taken = iter.take(rowSetSize)
+            val taken = iter.take(rowSetSize).toIndexedSeq.asInstanceOf[Seq[Row]]
             RowSet.toTRowSet(
-              taken.toIndexedSeq.asInstanceOf[Seq[Row]],
+              taken,
               resultSchema,
               getProtocolVersion)
           }