From f9e3086b2fa7eae24f22aa2fd32eb644d829e52b Mon Sep 17 00:00:00 2001
From: Holden Karau <holden@us.ibm.com>
Date: Tue, 19 Jan 2016 15:01:37 -0800
Subject: [PATCH] Round to one digit

---
 python/pyspark/ml/feature.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/pyspark/ml/feature.py b/python/pyspark/ml/feature.py
index 8365ba6214e09..837eaf623269f 100644
--- a/python/pyspark/ml/feature.py
+++ b/python/pyspark/ml/feature.py
@@ -1009,8 +1009,8 @@ class QuantileDiscretizer(JavaEstimator, HasInputCol, HasOutputCol):
     >>> splits = bucketizer.getSplits()
     >>> splits[0]
     -inf
-    >>> int(splits[1]*10)
-    4
+    >>> round(splits[1], 1)
+    0.4
     >>> bucketed = bucketizer.transform(df).head()
     >>> bucketed.buckets
     0.0