mirror of
https://github.com/wassname/ray.git
synced 2026-06-30 21:11:24 +08:00
libraylib -> lib (#311)
This commit is contained in:
committed by
Robert Nishihara
parent
c3695099fc
commit
4c323d4ecf
@@ -339,7 +339,7 @@ class Worker(object):
|
||||
"""
|
||||
if ray.lib.is_arrow(self.handle, objref):
|
||||
## this is the new codepath
|
||||
buff, segmentid, metadata_offset = ray.libraylib.get_buffer(self.handle, objref)
|
||||
buff, segmentid, metadata_offset = ray.lib.get_buffer(self.handle, objref)
|
||||
metadata_size = np.frombuffer(buff, dtype="int64", count=1)[0]
|
||||
metadata = np.frombuffer(buff, dtype="byte", offset=8, count=metadata_size)
|
||||
data = np.frombuffer(buff, dtype="byte")[8 + metadata_size:]
|
||||
|
||||
+3
-3
@@ -137,10 +137,10 @@ class ObjStoreTest(unittest.TestCase):
|
||||
|
||||
# Getting a buffer after modifying it before it finishes should return updated buffer
|
||||
objref = ray.lib.get_objref(w1.handle)
|
||||
buf = ray.libraylib.allocate_buffer(w1.handle, objref, 100)
|
||||
buf = ray.lib.allocate_buffer(w1.handle, objref, 100)
|
||||
buf[0][0] = 1
|
||||
ray.libraylib.finish_buffer(w1.handle, objref, buf[1], 0)
|
||||
completedbuffer = ray.libraylib.get_buffer(w1.handle, objref)
|
||||
ray.lib.finish_buffer(w1.handle, objref, buf[1], 0)
|
||||
completedbuffer = ray.lib.get_buffer(w1.handle, objref)
|
||||
self.assertEqual(completedbuffer[0][0], 1)
|
||||
|
||||
ray.services.cleanup()
|
||||
|
||||
Reference in New Issue
Block a user