fix style

apache · Jun 7, 2019 · db953d9 · db953d9
1 parent 27736c5
commit db953d9
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/core/src/test/scala/org/apache/spark/scheduler/TaskSchedulerImplSuite.scala b/core/src/test/scala/org/apache/spark/scheduler/TaskSchedulerImplSuite.scala
@@ -30,7 +30,6 @@ import org.scalatest.mockito.MockitoSugar
 
 import org.apache.spark._
 import org.apache.spark.ResourceUtils._
-import org.apache.spark.internal.config._
 import org.apache.spark.internal.Logging
 import org.apache.spark.internal.config
 import org.apache.spark.util.ManualClock
@@ -1250,8 +1249,8 @@ class TaskSchedulerImplSuite extends SparkFunSuite with LocalSparkContext with B
     val taskGpus = 1
     val executorGpus = 4
     val executorCpus = 4
-    val gpuTaskResourceID = ResourceID(SPARK_TASK_PREFIX, GPU)
-    val gpuExecResourceID = ResourceID(SPARK_EXECUTOR_PREFIX, GPU)
+    val gpuTaskResourceID = ResourceID(config.SPARK_TASK_PREFIX, GPU)
+    val gpuExecResourceID = ResourceID(config.SPARK_EXECUTOR_PREFIX, GPU)
 
     val taskScheduler = setupScheduler(numCores = executorCpus,
       config.CPUS_PER_TASK.key -> taskCpus.toString,