From 27d72585dd0fd4ffb042c6687aa668c88550be2a Mon Sep 17 00:00:00 2001 From: Andy Grove Date: Mon, 15 Jun 2020 12:41:37 -0600 Subject: [PATCH] Remove dependency on shaded Spark artifact (#169) --- .../test/scala/ai/rapids/spark/GpuCoalesceBatchesSuite.scala | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/src/test/scala/ai/rapids/spark/GpuCoalesceBatchesSuite.scala b/tests/src/test/scala/ai/rapids/spark/GpuCoalesceBatchesSuite.scala index ff4493f43f9..5a6f9d2d276 100644 --- a/tests/src/test/scala/ai/rapids/spark/GpuCoalesceBatchesSuite.scala +++ b/tests/src/test/scala/ai/rapids/spark/GpuCoalesceBatchesSuite.scala @@ -17,8 +17,7 @@ package ai.rapids.spark import java.io.File - -import org.sparkproject.guava.io.Files +import java.nio.file.Files import org.apache.spark.sql.execution.metric.SQLMetric import org.apache.spark.sql.rapids.metrics.source.MockTaskContext @@ -166,7 +165,7 @@ class GpuCoalesceBatchesSuite extends SparkQueryCompareTestSuite { .set("spark.rapids.sql.exec.FileSourceScanExec", "false") // force Parquet read onto CPU .set("spark.sql.shuffle.partitions", "1") - val dir = Files.createTempDir() + val dir = Files.createTempDirectory("spark-rapids-test").toFile val path = new File(dir, s"HostColumnarToGpu-${System.currentTimeMillis()}.parquet").getAbsolutePath