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

[Serve] Remove register_custom_serializer (#10331)

Browse Source
This commit is contained in:
Simon Mo
2020-08-25 21:20:43 -07:00
committed by GitHub
parent cbd9632f3a
commit ed3fdd2c0b
2 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/serve/router.py
+4 -1
View File
@@ -37,13 +37,16 @@ class Query:
self.shard_key = shard_key
self.is_shadow_query = is_shadow_query
def __reduce__(self):
return type(self).ray_deserialize, (self.ray_serialize(), )
def ray_serialize(self):
# NOTE: this method is needed because Query need to be serialized and
# sent to the replica worker. However, after we send the query to
# replica worker the async_future is still needed to retrieve the final
# result. Therefore we need a way to pass the information to replica
# worker without removing async_future.
clone = copy.copy(self).__dict__
clone = copy.copy(self.__dict__)
clone.pop("async_future")
return pickle.dumps(clone)