make sure it's iterator

Davies Liu · Davies Liu · commit af3713a1ee73 · 2015-04-09T14:36:24.000-07:00
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):
