wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-01 23:33:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

get rid of numbuf sudo command

Browse Source
This commit is contained in:
Philipp Moritz
2016-06-25 13:28:33 -07:00
parent abaa4211d4
commit dc70c90c2e
3 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/python/ray/serialization.py
+3
View File
@@ -47,6 +47,9 @@ def from_primitive(primitive_obj):
obj.deserialize(primitive_obj[1])
return obj
def is_arrow_serializable(value):
return type(value) == np.ndarray and value.dtype.name in ["int8", "int16", "int32", "int64", "uint8", "uint16", "uint32", "uint64", "float32", "float64"]
def serialize(worker_capsule, obj):
primitive_obj = to_primitive(obj)
obj_capsule, contained_objrefs = ray.lib.serialize_object(worker_capsule, primitive_obj) # contained_objrefs is a list of the objrefs contained in obj