From d9670f84739b0840501b19b8cb0e851850edb8c1 Mon Sep 17 00:00:00 2001 From: Cheng Hao Date: Wed, 16 Mar 2016 11:20:15 -0700 Subject: [SPARK-13894][SQL] SqlContext.range return type from DataFrame to DataSet ## What changes were proposed in this pull request? https://issues.apache.org/jira/browse/SPARK-13894 Change the return type of the `SQLContext.range` API from `DataFrame` to `Dataset`. ## How was this patch tested? No additional unit test required. Author: Cheng Hao Closes #11730 from chenghao-intel/range. --- .../src/test/scala/org/apache/spark/ml/feature/StringIndexerSuite.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'mllib') diff --git a/mllib/src/test/scala/org/apache/spark/ml/feature/StringIndexerSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/feature/StringIndexerSuite.scala index b9533f881d..d40e69dced 100644 --- a/mllib/src/test/scala/org/apache/spark/ml/feature/StringIndexerSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/ml/feature/StringIndexerSuite.scala @@ -114,7 +114,7 @@ class StringIndexerSuite val indexerModel = new StringIndexerModel("indexer", Array("a", "b", "c")) .setInputCol("label") .setOutputCol("labelIndex") - val df = sqlContext.range(0L, 10L) + val df = sqlContext.range(0L, 10L).toDF() assert(indexerModel.transform(df).eq(df)) } -- cgit v1.2.3