update bisecting k-means unit tests

apache · Jan 20, 2016 · 3a70c82 · 3a70c82
1 parent 4251780
commit 3a70c82
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/pyspark/mllib/tests.py b/python/pyspark/mllib/tests.py
@@ -426,7 +426,7 @@ def test_bisecting_kmeans(self):
         from pyspark.mllib.clustering import BisectingKMeans
         data = array([0.0, 0.0, 1.0, 1.0, 9.0, 8.0, 8.0, 9.0]).reshape(4, 2)
         bskm = BisectingKMeans()
-        model = bskm.train(sc.parallelize(data, 2), k=4)
+        model = bskm.train(self.sc.parallelize(data, 2), k=4)
         p = array([0.0, 0.0])
         rdd_p = self.sc.parallelize([p])
         self.assertEqual(model.predict(p), model.predict(rdd_p).first())