Try to fix the failing test (#356)

salesforce · Jul 9, 2019 · 3b82d96 · 3b82d96
1 parent c25f4eb
commit 3b82d96
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/features/src/test/scala/com/salesforce/op/stages/base/sequence/SequenceEstimatorTest.scala b/features/src/test/scala/com/salesforce/op/stages/base/sequence/SequenceEstimatorTest.scala
@@ -73,14 +73,14 @@ class FractionOfResponsesEstimator(uid: String = UID[FractionOfResponsesEstimato
     import dataset.sparkSession.implicits._
     val sizes = dataset.map(_.map(_.size))
     val size = getInputFeatures().length
-    val counts = sizes.select(SequenceAggregators.SumNumSeq[Int](size = size).toColumn).first()
+    val counts = sizes.select(SequenceAggregators.SumNumSeq[Int](size = size).toColumn).first().map(_.toDouble)
     new FractionOfResponsesModel(counts = counts, operationName = operationName, uid = uid)
   }
 }
 
 final class FractionOfResponsesModel private[op]
 (
-  val counts: Seq[Int],
+  val counts: Seq[Double],
   operationName: String,
   uid: String
 ) extends SequenceModel[DateList, OPVector](operationName = operationName, uid = uid) {