Allow any ArrowReader implementation to be use for reading Arrow data

Kotlin#627
fb64 · Mar 16, 2024 · 8ca2ed0 · 8ca2ed0
1 parent 35108fa
commit 8ca2ed0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataframe-arrow/src/main/kotlin/org/jetbrains/kotlinx/dataframe/io/arrowReadingImpl.kt b/dataframe-arrow/src/main/kotlin/org/jetbrains/kotlinx/dataframe/io/arrowReadingImpl.kt
@@ -317,7 +317,7 @@ internal fun DataFrame.Companion.readArrowImpl(
                         add(df)
                     }
                 }
-                is ArrowStreamReader -> {
+                else -> {
                     val root = reader.vectorSchemaRoot
                     val schema = root.schema
                     while (reader.loadNextBatch()) {