Address comments.

apache · Feb 15, 2018 · 174c114 · 174c114
1 parent 3a29039
commit 174c114
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/ml/feature/Bucketizer.scala b/mllib/src/main/scala/org/apache/spark/ml/feature/Bucketizer.scala
@@ -218,7 +218,6 @@ final class Bucketizer @Since("1.4.0") (@Since("1.4.0") override val uid: String
     defaultCopy[Bucketizer](extra).setParent(parent)
   }
 
-  @Since("2.3.0")
   override def write: MLWriter = new Bucketizer.BucketizerWriter(this)
 }
 
@@ -309,7 +308,7 @@ object Bucketizer extends DefaultParamsReadable[Bucketizer] {
       // we can get rid of this.
       var paramWithoutOutputCol: Option[JValue] = None
       if (instance.isSet(instance.inputCols)) {
-        val params = instance.extractParamMap().toSeq.asInstanceOf[Seq[ParamPair[Any]]]
+        val params = instance.extractParamMap().toSeq
         val jsonParams = params.filter(_.param != instance.outputCol).map { case ParamPair(p, v) =>
           p.name -> parse(p.jsonEncode(v))
         }.toList

diff --git a/mllib/src/main/scala/org/apache/spark/ml/feature/QuantileDiscretizer.scala b/mllib/src/main/scala/org/apache/spark/ml/feature/QuantileDiscretizer.scala
@@ -254,7 +254,6 @@ final class QuantileDiscretizer @Since("1.6.0") (@Since("1.6.0") override val ui
   @Since("1.6.0")
   override def copy(extra: ParamMap): QuantileDiscretizer = defaultCopy(extra)
 
-  @Since("2.3.0")
   override def write: MLWriter = new QuantileDiscretizer.QuantileDiscretizerWriter(this)
 }
 
@@ -273,7 +272,7 @@ object QuantileDiscretizer extends DefaultParamsReadable[QuantileDiscretizer] wi
       // we can get rid of this.
       var paramWithoutOutputCol: Option[JValue] = None
       if (instance.isSet(instance.inputCols)) {
-        val params = instance.extractParamMap().toSeq.asInstanceOf[Seq[ParamPair[Any]]]
+        val params = instance.extractParamMap().toSeq
         val jsonParams = params.filter(_.param != instance.outputCol).map { case ParamPair(p, v) =>
           p.name -> parse(p.jsonEncode(v))
         }.toList