Setting the default needs to come after the value gets defined

apache · Jul 31, 2015 · 6a38edb · 6a38edb
1 parent 9e241d8
commit 6a38edb
Showing 1 changed file with 1 addition and 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
@@ -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