aboutsummaryrefslogtreecommitdiff
path: root/mllib/src/test
diff options
context:
space:
mode:
authorCheng Lian <lian@databricks.com>2016-03-13 12:02:52 +0800
committerCheng Lian <lian@databricks.com>2016-03-13 12:02:52 +0800
commitc079420d7c55d8972db716a2695a5ddd606d11cd (patch)
treecf33f0c9895ff28d2ce686c210d0d4afd7f6b78d /mllib/src/test
parent4eace4d384f0e12b4934019d8654b5e3886ddaef (diff)
downloadspark-c079420d7c55d8972db716a2695a5ddd606d11cd.tar.gz
spark-c079420d7c55d8972db716a2695a5ddd606d11cd.tar.bz2
spark-c079420d7c55d8972db716a2695a5ddd606d11cd.zip
[SPARK-13841][SQL] Removes Dataset.collectRows()/takeRows()
## What changes were proposed in this pull request? This PR removes two methods, `collectRows()` and `takeRows()`, from `Dataset[T]`. These methods were added in PR #11443, and were later considered not useful. ## How was this patch tested? Existing tests should do the work. Author: Cheng Lian <lian@databricks.com> Closes #11678 from liancheng/remove-collect-rows-and-take-rows.
Diffstat (limited to 'mllib/src/test')
-rw-r--r--mllib/src/test/java/org/apache/spark/ml/feature/JavaVectorSlicerSuite.java2
1 files changed, 1 insertions, 1 deletions
diff --git a/mllib/src/test/java/org/apache/spark/ml/feature/JavaVectorSlicerSuite.java b/mllib/src/test/java/org/apache/spark/ml/feature/JavaVectorSlicerSuite.java
index b87605ebfd..e2da11183b 100644
--- a/mllib/src/test/java/org/apache/spark/ml/feature/JavaVectorSlicerSuite.java
+++ b/mllib/src/test/java/org/apache/spark/ml/feature/JavaVectorSlicerSuite.java
@@ -78,7 +78,7 @@ public class JavaVectorSlicerSuite {
Dataset<Row> output = vectorSlicer.transform(dataset);
- for (Row r : output.select("userFeatures", "features").takeRows(2)) {
+ for (Row r : output.select("userFeatures", "features").takeAsList(2)) {
Vector features = r.getAs(1);
Assert.assertEquals(features.size(), 2);
}