fix tests

markhamstra · Oct 11, 2014 · 64561e4 · 64561e4
1 parent 331ecce
commit 64561e4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/pyspark/streaming/tests.py b/python/pyspark/streaming/tests.py
@@ -487,9 +487,8 @@ def test_get_or_create(self):
         inputd = tempfile.mkdtemp()
         outputd = tempfile.mkdtemp() + "/"
 
-        def updater(it):
-            for k, vs, s in it:
-                yield (k, sum(vs, s or 0))
+        def updater(vs, s):
+            return sum(vs, s or 0)
 
         def setup():
             conf = SparkConf().set("spark.default.parallelism", 1)