fix r / w hadoop methods

Signed-off-by: Grigory Pomadchin <gr.pomadchin@gmail.com>
locationtech · Apr 3, 2017 · c7547ff · c7547ff
1 parent db86ac5
commit c7547ff
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/spark/src/main/scala/geotrellis/spark/io/hadoop/HadoopRasterMethods.scala b/spark/src/main/scala/geotrellis/spark/io/hadoop/HadoopRasterMethods.scala
@@ -19,11 +19,11 @@ package geotrellis.spark.io.hadoop
 import geotrellis.util.MethodExtensions
 
 import org.apache.hadoop.conf.Configuration
-import org.apache.hadoop.fs.{FileSystem, Path}
+import org.apache.hadoop.fs.Path
 import org.apache.hadoop.io.compress.CompressionCodecFactory
 import org.apache.spark.SparkContext
-import java.io.{DataInputStream, DataOutputStream}
 
+import java.io.{DataInputStream, DataOutputStream}
 
 trait HadoopRasterMethods[T] extends MethodExtensions[T] {
   def write(path: Path)(implicit sc: SparkContext): Unit = write(path, sc.hadoopConfiguration)
@@ -32,7 +32,7 @@ trait HadoopRasterMethods[T] extends MethodExtensions[T] {
 
 object HadoopRasterMethods {
   def write(path: Path, conf: Configuration)(dosWrite: DataOutputStream => Unit): Unit = {
-    val fs = FileSystem.get(conf)
+    val fs = path.getFileSystem(conf)
 
     val os = {
       val factory = new CompressionCodecFactory(conf)
@@ -58,7 +58,7 @@ object HadoopRasterMethods {
   }
 
   def read[T](path: Path, conf: Configuration)(disRead: DataInputStream => T): T = {
-    val fs = FileSystem.get(conf)
+    val fs = path.getFileSystem(conf)
 
     val is = {
       val factory = new CompressionCodecFactory(conf)