Round to one digit

apache · Jan 19, 2016 · f9e3086 · f9e3086
1 parent d90339a
commit f9e3086
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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