diff options
Diffstat (limited to 'python/pyspark/resultiterable.py')
-rw-r--r-- | python/pyspark/resultiterable.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/python/pyspark/resultiterable.py b/python/pyspark/resultiterable.py index 7f418f8d2e..df34740fc8 100644 --- a/python/pyspark/resultiterable.py +++ b/python/pyspark/resultiterable.py @@ -19,6 +19,7 @@ __all__ = ["ResultIterable"] import collections + class ResultIterable(collections.Iterable): """ A special result iterable. This is used because the standard iterator can not be pickled @@ -27,7 +28,9 @@ class ResultIterable(collections.Iterable): self.data = data self.index = 0 self.maxindex = len(data) + def __iter__(self): return iter(self.data) + def __len__(self): return len(self.data) |