[Serve] Improve Serialization (#7688)

This commit is contained in:
Simon Mo
2020-03-29 14:57:19 -07:00
committed by GitHub
parent fc23f79f82
commit 353d7e107f
5 changed files with 34 additions and 27 deletions
+4
View File
@@ -17,6 +17,7 @@ from ray.serve.exceptions import RayServeException, batch_annotation_not_found
from ray.serve.backend_config import BackendConfig
from ray.serve.policy import RoutePolicy
from ray.serve.queues import Query
from ray.serve.request_params import RequestMetadata
global_state = None
@@ -124,6 +125,9 @@ def init(
# Register serialization context once
ray.register_custom_serializer(Query, Query.ray_serialize,
Query.ray_deserialize)
ray.register_custom_serializer(RequestMetadata,
RequestMetadata.ray_serialize,
RequestMetadata.ray_deserialize)
if kv_store_path is None:
_, kv_store_path = mkstemp()