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

Re-route asyncio plasma code path through raylet instead of direct plasma connection (#7234)

Browse Source
This commit is contained in:
ijrsvt
2020-03-03 15:43:46 -05:00
committed by GitHub
parent c2c6d96490
commit fb76092d75
14 changed files with 169 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/_raylet.pyx
+5 -2
View File
@@ -635,9 +635,12 @@ cdef class CoreWorker:
def set_actor_title(self, title):
self.core_worker.get().SetActorTitle(title)
def subscribe_to_plasma(self, plasma_event_handler):
def set_plasma_added_callback(self, plasma_event_handler):
self.plasma_event_handler = plasma_event_handler
self.core_worker.get().SubscribeToAsyncPlasma(async_plasma_callback)
self.core_worker.get().SetPlasmaAddedCallback(async_plasma_callback)
def subscribe_to_plasma_object(self, ObjectID object_id):
self.core_worker.get().SubscribeToPlasmaAdd(object_id.native())
def get_plasma_event_handler(self):
return self.plasma_event_handler