From e2dbe15fa945a63a045558ad8f10239510b166f2 Mon Sep 17 00:00:00 2001 From: Holden Karau Date: Fri, 30 Mar 2018 14:26:26 -0700 Subject: [PATCH] Call testPMMLWrite to actually verify PMML export --- .../test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala | 1 + 1 file changed, 1 insertion(+) diff --git a/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala index 082e23110fdda..40b13a5d3cc99 100644 --- a/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala @@ -226,6 +226,7 @@ class KMeansSuite extends SparkFunSuite with MLlibTestSparkContext with DefaultR val pmmlClusteringModel = pmml.getModels.get(0).asInstanceOf[ClusteringModel] assert(pmmlClusteringModel.getNumberOfClusters === clusterCenters.length) } + testPMMLWrite(sc, model, checkModel) } }