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

Simplify Cython ID types and fix bug of ActorCheckpointID (#4045)

Browse Source
This commit is contained in:
Hao Chen
2019-02-15 20:15:16 +08:00
committed by GitHub
parent bb7c4ce9c4
commit 042ad84573
9 changed files with 108 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/worker.py
+1 -1
View File
@@ -1128,7 +1128,7 @@ def _initialize_serialization(driver_id, worker=global_worker):
serialization_context.set_pickle(pickle.dumps, pickle.loads)
pyarrow.register_torch_serialization_handlers(serialization_context)
for id_type in ray._ID_TYPES:
for id_type in ray._raylet._ID_TYPES:
serialization_context.register_type(
id_type,
"{}.{}".format(id_type.__module__, id_type.__name__),