From 3b82d9683bfa851833e0e0ec0a52b73b304ae5e0 Mon Sep 17 00:00:00 2001 From: Matthew Tovbin Date: Tue, 9 Jul 2019 09:38:39 -0700 Subject: [PATCH] Try to fix the failing test (#356) --- .../op/stages/base/sequence/SequenceEstimatorTest.scala | 4 ++-- 1 file changed, 2 insertions(+), 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 index 427fb8f55a..bfb926dcd8 100644 --- 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) {