make sure it's iterator

nemccarthy · Apr 9, 2015 · af3713a · af3713a
1 parent 67772dd
commit af3713a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/pyspark/shuffle.py b/python/pyspark/shuffle.py
@@ -659,7 +659,7 @@ class GroupByKey(object):
     """
 
     def __init__(self, iterator):
-        self.iterator = iterator
+        self.iterator = iter(iterator)
         self.next_item = None
 
     def __iter__(self):