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

Properly mock ray submodules when building documentation. (#337)

Browse Source
This commit is contained in:
Robert Nishihara
2017-03-04 23:02:56 -08:00
committed by Philipp Moritz
parent 0a233b7144
commit a7ddac6fb1
8 changed files with 74 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/serialization.py
+9 -3
View File
@@ -4,7 +4,7 @@ from __future__ import print_function
import numpy as np
import ray.numbuf as numbuf
import ray.numbuf
import ray.pickling as pickling
def check_serializable(cls):
@@ -139,5 +139,11 @@ def deserialize(serialized_obj):
obj.__dict__.update(serialized_obj)
return obj
# Register the callbacks with numbuf.
numbuf.register_callbacks(serialize, deserialize)
def set_callbacks():
"""Register the custom callbacks with numbuf.
The serialize callback is used to serialize objects that numbuf does not know
how to serialize (for example custom Python classes). The deserialize callback
is used to serialize objects that were serialized by the serialize callback.
"""
ray.numbuf.register_callbacks(serialize, deserialize)