o ci @sdZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddl mZer||jj |<||jj |jvrQ|jj |j |td|j|j dddSdS)NObject)typer ref_idnamebaseline_optionsActor) isinstanceray ObjectRefbinaryr object_refsr ractor ActorHandle _actor_id actor_refs actor_ownersadd)robjobj_idactor_idrrr persistent_id&s2    zServerPickler.persistent_id)__name__ __module__ __qualname__strrr* __classcell__rrrrr sr cs$eZdZfddZddZZS)ClientUnpicklercstj|i|||_dSr )r rr )rr rrrrrrFs zClientUnpickler.__init__cCst|tsJ|jdkrtS|jdkr|jj|j|jS|jdkr)|jj|jS|jdkr5t |j|jS|jdkrE|j |j|j|j S|jdkrQt |j|jS|jdkra|j |j|j|j S|jdkrs|jj|j}t||jStd ) NRayrrRemoteFuncSelfReference RemoteFuncRemoteActorSelfReference RemoteActor RemoteMethodzUncovered client data type)rrrrr r r rr$rlookup_or_register_funcrrlookup_or_register_actorgetattrrNotImplementedError)rpidr!rrrpersistent_loadJs.           zClientUnpickler.persistent_load)r+r,r-rr<r/rrrrr0Es r0r'r server_instancerreturncCsNt}t||||d}|||WdS1s wYdS)N)protocol)ioBytesIOr dumpgetvalue)r'r r=r?filesprrrdumps_from_serveres  $rFTASCIIstrict) fix_importsencodingerrorsdatacCsZt t|tr tdt|}t||||dWdS1s&wYdS)Nz%Can't load pickle from unicode string)rIrJ)rrr. TypeErrorr@rAr0load)rLr=rIrJrKrDrrrloads_from_clientns  $rOr )__doc__r@rtypingrrray._private.client_mode_hookrray.cloudpickle cloudpickleray.util.client.client_picklerr#ray.util.client.server.server_stubsrrray.util.client.server.serverrpickle CloudPicklerr Unpicklerr0r.bytesrFrOrrrrsF         %!