From 6a38edb1ec5716d6f21fdb2c7ed7b39d1ea53a2e Mon Sep 17 00:00:00 2001 From: Holden Karau Date: Fri, 31 Jul 2015 00:04:23 -0700 Subject: [PATCH] Setting the default needs to come after the value gets defined --- .../main/scala/org/apache/spark/ml/feature/StringIndexer.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mllib/src/main/scala/org/apache/spark/ml/feature/StringIndexer.scala b/mllib/src/main/scala/org/apache/spark/ml/feature/StringIndexer.scala index e7806d859baa1..a406454a7e6ab 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/feature/StringIndexer.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/feature/StringIndexer.scala @@ -194,7 +194,6 @@ class StringIndexerInverse private[ml] ( * @group setParam */ def setLabels(value: Array[String]): this.type = set(labels, value) - setDefault(labels -> Array.empty[String]) /** * Param for array of labels. @@ -203,6 +202,7 @@ class StringIndexerInverse private[ml] ( * @group param */ final val labels: StringArrayParam = new StringArrayParam(this, "labels", "array of labels") + setDefault(labels, Array.empty[String]) /** @group getParam * Optional labels to be provided by the user, if not supplied column