Merge pull request #13784 from DevinTDHa/bug/dbfs

Fix loadSavedModel for DBFS
JohnSnowLabs · May 10, 2023 · b8a8cb1 · b8a8cb1
2 parents 2153a06 + d30652d
commit b8a8cb1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/scala/com/johnsnowlabs/nlp/util/io/ResourceHelper.scala b/src/main/scala/com/johnsnowlabs/nlp/util/io/ResourceHelper.scala
@@ -106,7 +106,7 @@ object ResourceHelper {
   case class SourceStream(resource: String) {
 
     var fileSystem: Option[FileSystem] = None
-    private val (pathExists, path) = OutputHelper.doesPathExists(resource)
+    private val (pathExists: Boolean, path: Option[Path]) = OutputHelper.doesPathExists(resource)
     if (!pathExists) {
       throw new FileNotFoundException(s"file or folder: $resource not found")
     } else {
@@ -156,7 +156,7 @@ object ResourceHelper {
             Paths.get(destination.toString, path.get.getName).toUri
           else destination.toUri
         case "dbfs" =>
-          val dbfsPath = path.toString.replace("dbfs:/", "/dbfs/")
+          val dbfsPath = path.get.toString.replace("dbfs:/", "/dbfs/")
           val sourceFile = new File(dbfsPath)
           val targetFile = new File(destination.toString)
           if (sourceFile.isFile) FileUtils.copyFileToDirectory(sourceFile, targetFile)