wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-09-12 12:51:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Migrate Python C extension to Cython (#3541)

Browse Source
This commit is contained in:
Si-Yuan
2019-01-24 09:17:14 -08:00
committed by Philipp Moritz
parent c1a52b1c86
commit 48139cf861
49 changed files with 1909 additions and 1827 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/utils/actors.py
+2 -2
View File
@@ -39,14 +39,14 @@ class TaskPool(object):
Assumes obj_id only is one id."""
for worker, obj_id in self.completed():
plasma_id = ray.pyarrow.plasma.ObjectID(obj_id.id())
plasma_id = ray.pyarrow.plasma.ObjectID(obj_id.binary())
(ray.worker.global_worker.raylet_client.fetch_or_reconstruct(
[obj_id], True))
self._fetching.append((worker, obj_id))
remaining = []
for worker, obj_id in self._fetching:
plasma_id = ray.pyarrow.plasma.ObjectID(obj_id.id())
plasma_id = ray.pyarrow.plasma.ObjectID(obj_id.binary())
if ray.worker.global_worker.plasma_client.contains(plasma_id):
yield (worker, obj_id)
else: