fixing dataframe map to rdd.map

apache · Mar 8, 2016 · 1c9b9cd · 1c9b9cd
1 parent 1a486f8
commit 1c9b9cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mllib/src/test/scala/org/apache/spark/ml/feature/OneHotEncoderSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/feature/OneHotEncoderSuite.scala
@@ -130,7 +130,7 @@ class OneHotEncoderSuite
         .setDropLast(false)
       val encoded = encoder.transform(dfWithTypes)
 
-      val output = encoded.select("id", "labelVec").map { r =>
+      val output = encoded.select("id", "labelVec").rdd.map { r =>
         val vec = r.getAs[Vector](1)
         (r.getInt(0), vec(0), vec(1), vec(2))
       }.collect().toSet