diff --git a/istarmap.py b/istarmap.py index 8a55bdb..1edc7b1 100644 --- a/istarmap.py +++ b/istarmap.py @@ -14,7 +14,7 @@ def istarmap(self, func, iterable, chunksize=1): chunksize)) task_batches = mpp.Pool._get_tasks(func, iterable, chunksize) - result = mpp.IMapIterator(self._cache) + result = mpp.IMapIterator(self) self._taskqueue.put( ( self._guarded_task_generation(result._job, @@ -25,4 +25,4 @@ def istarmap(self, func, iterable, chunksize=1): return (item for chunk in result for item in chunk) -mpp.Pool.istarmap = istarmap \ No newline at end of file +mpp.Pool.istarmap = istarmap