Improved multi pick list map vectorizer test (#292)

salesforce · Apr 17, 2019 · f984f21 · f984f21
1 parent f228f26
commit f984f21
Show file tree

Hide file tree

Showing 2 changed files with 91 additions and 86 deletions.
diff --git a/core/src/main/scala/com/salesforce/op/stages/impl/feature/MultiPickListMapVectorizer.scala b/core/src/main/scala/com/salesforce/op/stages/impl/feature/MultiPickListMapVectorizer.scala
@@ -51,7 +51,7 @@ class MultiPickListMapVectorizer[T <: OPMap[Set[String]]]
   uid: String = UID[MultiPickListMapVectorizer[T]]
 )(implicit tti: TypeTag[T], ttiv: TypeTag[T#Value])
   extends SequenceEstimator[T, OPVector](operationName = "vecCatMap", uid = uid)
-    with VectorizerDefaults with PivotParams with MapPivotParams with TextParams
+    with PivotParams with MapPivotParams with TextParams
     with MapStringPivotHelper with CleanTextMapFun with MinSupportParam with TrackNullsParam
     with MaxPctCardinalityParams with MaxPctCardinalityFun {
 
@@ -94,8 +94,7 @@ final class MultiPickListMapVectorizerModel[T <: OPMap[Set[String]]] private[op]
   operationName: String,
   uid: String
 )(implicit tti: TypeTag[T])
-  extends SequenceModel[T, OPVector](operationName = operationName, uid = uid)
-    with VectorizerDefaults with CleanTextMapFun {
+  extends SequenceModel[T, OPVector](operationName = operationName, uid = uid) with CleanTextMapFun {
 
   def transformFn: (Seq[T]) => OPVector = row => {
     // Combine top values for each feature with map feature