Skip to content

Commit 954d7b9

Browse files
committed
fix style error
1 parent 99d46ee commit 954d7b9

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

mllib/src/test/scala/org/apache/spark/ml/feature/QuantileDiscretizerSuite.scala

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,14 +17,12 @@
1717

1818
package org.apache.spark.ml.feature
1919

20-
import org.scalatest.FunSuite
21-
22-
import org.apache.spark.SparkContext
2320
import org.apache.spark.ml.attribute.{Attribute, NominalAttribute}
2421
import org.apache.spark.mllib.util.MLlibTestSparkContext
2522
import org.apache.spark.sql.{Row, SQLContext}
23+
import org.apache.spark.{SparkContext, SparkFunSuite}
2624

27-
class QuantileDiscretizerSuite extends FunSuite with MLlibTestSparkContext {
25+
class QuantileDiscretizerSuite extends SparkFunSuite with MLlibTestSparkContext {
2826
import org.apache.spark.ml.feature.QuantileDiscretizerSuite._
2927

3028
test("Test quantile discretizer") {
@@ -55,7 +53,7 @@ class QuantileDiscretizerSuite extends FunSuite with MLlibTestSparkContext {
5553
}
5654
}
5755

58-
private object QuantileDiscretizerSuite extends FunSuite {
56+
private object QuantileDiscretizerSuite extends SparkFunSuite {
5957

6058
def checkDiscretizedData(
6159
sc: SparkContext,

0 commit comments

Comments
 (0)