Skip to content

Commit d82575f

Browse files
committed
fix test
1 parent 700e70f commit d82575f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ class StringIndexerSuite extends FunSuite with MLlibTestSparkContext {
3535
val data = sc.parallelize(Seq((0, "a"), (1, "b"), (2, "c"), (3, "a"), (4, "a"), (5, "c")), 2)
3636
val df = sqlContext.createDataFrame(data).toDF("id", "label")
3737
val indexer = new StringIndexer()
38+
.setInputCol("label")
3839
.setOutputCol("labelIndex")
3940
.fit(df)
4041
val transformed = indexer.transform(df)

0 commit comments

Comments
 (0)