mirror of
https://github.com/wassname/ray.git
synced 2026-07-12 21:55:59 +08:00
This reverts commit a68cda0a33.
This commit is contained in:
@@ -40,12 +40,16 @@ class TaskPool(object):
|
||||
Assumes obj_id only is one id."""
|
||||
|
||||
for worker, obj_id in self.completed(blocking_wait=blocking_wait):
|
||||
(ray.worker.global_worker.raylet_client.fetch_or_reconstruct(
|
||||
[obj_id], True))
|
||||
self._fetching.append((worker, obj_id))
|
||||
|
||||
remaining = []
|
||||
num_yielded = 0
|
||||
for worker, obj_id in self._fetching:
|
||||
if num_yielded < max_yield:
|
||||
if (num_yielded < max_yield
|
||||
and ray.worker.global_worker.core_worker.object_exists(
|
||||
obj_id)):
|
||||
yield (worker, obj_id)
|
||||
num_yielded += 1
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user