Tell Ray how to serialize FunctionSignature objects. (#1308)

2026-07-02 09:38:00 +08:00 · 2017-12-10 22:40:28 -08:00
parent 7009538321
commit 96c46d35ff
1 changed files with 4 additions and 0 deletions
@@ -1057,6 +1057,10 @@ def _initialize_serialization(worker=global_worker):
        # Ray can serialize actor handles that have been wrapped.
        register_custom_serializer(ray.actor.ActorHandleWrapper,
                                   use_dict=True)
+        # Tell Ray to serialize FunctionSignatures as dictionaries. This is
+        # used when passing around actor handles.
+        register_custom_serializer(ray.signature.FunctionSignature,
+                                   use_dict=True)


 def get_address_info_from_redis_helper(redis_address, node_ip_address):