Allow any ArrowReader implementation to be use for reading Arrow data

fb64 · fb64 · commit 2457bf846fa5 · 2024-03-18T19:36:59.000+01:00
#627
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()) {

Original file line number	Diff line number	Diff line change
`@@ -317,7 +317,7 @@ internal fun DataFrame.Companion.readArrowImpl(`
`317`	`317`	`add(df)`
`318`	`318`	`}`
`319`	`319`	`}`
`320`		`- is ArrowStreamReader -> {`
	`320`	`+ else -> {`
`321`	`321`	`val root = reader.vectorSchemaRoot`
`322`	`322`	`val schema = root.schema`
`323`	`323`	`while (reader.loadNextBatch()) {`