[SEDONA-696] Fix geopackage on databricks. (#1747)

apache · Jan 15, 2025 · cc69f28 · cc69f28
1 parent 52b6ae8
commit cc69f28
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/...5/src/main/scala/org/apache/sedona/sql/datasources/geopackage/GeoPackageScanBuilder.scala b/...5/src/main/scala/org/apache/sedona/sql/datasources/geopackage/GeoPackageScanBuilder.scala
@@ -37,13 +37,11 @@ class GeoPackageScanBuilder(
     extends FileScanBuilder(sparkSession, fileIndex, dataSchema) {
 
   override def build(): Scan = {
-    val paths = fileIndex.allFiles().map(_.getPath.toString)
-
     val fileIndexAdjusted =
       if (loadOptions.showMetadata)
         new InMemoryFileIndex(
           sparkSession,
-          paths.slice(0, 1).map(new org.apache.hadoop.fs.Path(_)),
+          fileIndex.inputFiles.slice(0, 1).map(new org.apache.hadoop.fs.Path(_)),
           options.asCaseSensitiveMap.asScala.toMap,
           userDefinedSchema)
       else fileIndex