wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 03:18:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Hotfix missing fields in multiprocessing.Pool (#6784)

Browse Source
This commit is contained in:
Edward Oakes
2020-01-13 16:39:33 -06:00
committed by GitHub
parent 4ee566129f
commit 3ea3b56eb1
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/experimental/multiprocessing/pool.py
+3 -1
View File
@@ -168,6 +168,8 @@ class IMapIterator:
"""Base class for OrderedIMapIterator and UnorderedIMapIterator."""
def __init__(self, pool, func, iterable, chunksize=None):
self._pool = pool
self._func = func
self._next_chunk_index = 0
# List of bools indicating if the given chunk is ready or not for all
# submitted chunks. Ordering mirrors that in the in the ResultThread.
@@ -182,7 +184,7 @@ class IMapIterator:
[], total_object_ids=self._total_chunks)
self._result_thread.start()
for _ in range(len(pool._actor_pool)):
for _ in range(len(self._pool._actor_pool)):
self._submit_next_chunk()
def _submit_next_chunk(self):