File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed
mllib/src/main/scala/org/apache/spark/ml/feature Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -351,9 +351,7 @@ object OneHotEncoderModel extends MLReadable[OneHotEncoderModel] {
351351 */
352352private [feature] object OneHotEncoderCommon {
353353
354- private def genOutputAttrNames (
355- inputCol : StructField ,
356- outputColName : String ): Option [Array [String ]] = {
354+ private def genOutputAttrNames (inputCol : StructField ): Option [Array [String ]] = {
357355 val inputAttr = Attribute .fromStructField(inputCol)
358356 inputAttr match {
359357 case nominal : NominalAttribute =>
@@ -399,7 +397,7 @@ private[feature] object OneHotEncoderCommon {
399397 inputCol : StructField ,
400398 dropLast : Boolean ,
401399 outputColName : String ): StructField = {
402- val outputAttrNames = genOutputAttrNames(inputCol, outputColName )
400+ val outputAttrNames = genOutputAttrNames(inputCol)
403401 val filteredOutputAttrNames = outputAttrNames.map { names =>
404402 if (dropLast) {
405403 require(names.length > 1 ,
You can’t perform that action at this time.
0 commit comments