chop of extra

holdenk · holdenk · commit 2afd197cf74c · 2015-12-01T18:47:11.000-08:00
diff --git a/python/pyspark/ml/feature.py b/python/pyspark/ml/feature.py
@@ -990,7 +990,7 @@ class QuantileDiscretizer(JavaEstimator, HasInputCol, HasOutputCol):
     >>> splits = bucketizer.getSplits()
     >>> splits[0]
     -inf
-    >>> splits[1]*10
+    >>> int(splits[1]*10)
     4
     >>> bucketed = bucketizer.transform(df).collect()
     >>> bucketed[0].buckets

Original file line number	Diff line number	Diff line change
`@@ -990,7 +990,7 @@ class QuantileDiscretizer(JavaEstimator, HasInputCol, HasOutputCol):`
`990`	`990`	`>>> splits = bucketizer.getSplits()`
`991`	`991`	`>>> splits[0]`
`992`	`992`	`-inf`
`993`		`- >>> splits[1]*10`
	`993`	`+ >>> int(splits[1]*10)`
`994`	`994`	`4`
`995`	`995`	`>>> bucketed = bucketizer.transform(df).collect()`
`996`	`996`	`>>> bucketed[0].buckets`