fix generator issue

markhamstra · Jun 24, 2015 · a0c0161 · a0c0161
1 parent 8a9ef9c
commit a0c0161
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/pyspark/rdd.py b/python/pyspark/rdd.py
@@ -711,9 +711,10 @@ def check_return_code():
                     raise Exception("Pipe function `%s' exited "
                                     "with error code %d" % (command, pipe.returncode))
                 else:
-                    yield None
+                    for i in range(0):
+                        yield i
             return (x.rstrip(b'\n').decode('utf-8') for x in
-                    chain(iter(pipe.stdout.readline, b''), iter(check_return_code, None)))
+                    chain(iter(pipe.stdout.readline, b''), check_return_code()))
         return self.mapPartitions(func)
 
     def foreach(self, f):