Compose functions via cats in GDAL tests

locationtech · Jun 27, 2022 · f3321b0 · f3321b0
1 parent bc133d2
commit f3321b0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gdal-spark/src/test/scala/geotrellis/spark/gdal/GDALRasterSourceRDDSpec.scala b/gdal-spark/src/test/scala/geotrellis/spark/gdal/GDALRasterSourceRDDSpec.scala
@@ -29,6 +29,8 @@ import geotrellis.spark.testkit._
 import geotrellis.store.hadoop._
 import cats.effect._
 import cats.syntax.parallel._
+import cats.syntax.functor._
+import cats.instances.function._
 import spire.syntax.cfor._
 import org.apache.spark.rdd.RDD
 import org.scalatest.Inspectors._
@@ -54,7 +56,7 @@ class GDALRasterSourceRDDSpec extends AnyFunSpec with TestEnvironment with Befor
     scheduler.setRemoveOnCancelPolicy(true)
     val (s, sh) = (unsafe.Scheduler.fromScheduledExecutor(scheduler), { () => scheduler.shutdown() })
 
-    unsafe.IORuntime(compute, blocking, s, () => { sh(); shutdown(); }, unsafe.IORuntimeConfig())
+    unsafe.IORuntime(compute, blocking, s, sh.as(shutdown()), unsafe.IORuntimeConfig())
   }
 
   implicit val runtime: unsafe.IORuntime = {