fix tests

apache · Nov 11, 2014 · d5057c4 · d5057c4
1 parent 6987611
commit d5057c4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/pyspark/mllib/feature.py b/python/pyspark/mllib/feature.py
@@ -104,7 +104,7 @@ def transform(self, vector):
             vector = vector.map(_convert_to_vector)
         else:
             vector = _convert_to_vector(vector)
-        return self.call("transform", dataset)
+        return self.call("transform", vector)
 
 
 class StandardScalerModel(JavaVectorTransformer):

diff --git a/python/pyspark/mllib/recommendation.py b/python/pyspark/mllib/recommendation.py
@@ -51,7 +51,7 @@ class MatrixFactorizationModel(JavaModelWrapper):
     >>> testset = sc.parallelize([(1, 2), (1, 1)])
     >>> model = ALS.train(ratings, 1, seed=10)
     >>> model.predictAll(testset).collect()
-    [Rating(1, 1, 1), Rating(1, 2, 1)]
+    [Rating(1, 1, 1.0471...), Rating(1, 2, 1.9679...)]
 
     >>> model = ALS.train(ratings, 4, seed=10)
     >>> model.userFeatures().collect()