def _put_object(self, request: ray_client_pb2.PutRequest, client_id: str, context=None): """Put an object in the cluster with ray.put() via gRPC. Args: request: PutRequest with pickled data. client_id: The client who owns this data, for tracking when to delete this reference. context: gRPC context. """ try: obj = loads_from_client(request.data, self) with disable_client_hook(): objectref = ray.put(obj) except Exception as e: logger.exception("Put failed:") return ray_client_pb2.PutResponse(id=b"", valid=False, error=cloudpickle.dumps(e)) self.object_refs[client_id][objectref.binary()] = objectref if len(request.client_ref_id) > 0: self.client_side_ref_map[client_id][ request.client_ref_id] = objectref.binary() logger.debug("put: %s" % objectref) return ray_client_pb2.PutResponse(id=objectref.binary(), valid=True)
def _put_object( self, data: Union[bytes, bytearray], client_ref_id: bytes, client_id: str, context=None, ): """Put an object in the cluster with ray.put() via gRPC. Args: data: Pickled data. Can either be bytearray if this is called from the dataservicer, or bytes if called from PutObject. client_ref_id: The id associated with this object on the client. client_id: The client who owns this data, for tracking when to delete this reference. context: gRPC context. """ try: obj = loads_from_client(data, self) with disable_client_hook(): objectref = ray.put(obj) except Exception as e: logger.exception("Put failed:") return ray_client_pb2.PutResponse(id=b"", valid=False, error=cloudpickle.dumps(e)) self.object_refs[client_id][objectref.binary()] = objectref if len(client_ref_id) > 0: self.client_side_ref_map[client_id][ client_ref_id] = objectref.binary() logger.debug("put: %s" % objectref) return ray_client_pb2.PutResponse(id=objectref.binary(), valid=True)
def _put_object(self, request: ray_client_pb2.PutRequest, client_id: str, context=None): """Put an object in the cluster with ray.put() via gRPC. Args: request: PutRequest with pickled data. client_id: The client who owns this data, for tracking when to delete this reference. context: gRPC context. """ obj = loads_from_client(request.data, self) objectref = ray.put(obj) self.object_refs[client_id][objectref.binary()] = objectref logger.debug("put: %s" % objectref) return ray_client_pb2.PutResponse(id=objectref.binary())
def PutObject(self, request, context=None): data = cloudpickle.dumps(request.data) objectref = ray.put(data) self.realref[objectref.binary()] = objectref print("put: %s" % objectref) return ray_client_pb2.PutResponse(id=objectref.binary())
def PutObject(self, request, context=None) -> ray_client_pb2.PutResponse: obj = cloudpickle.loads(request.data) objectref = self._put_and_retain_obj(obj) pickled_ref = cloudpickle.dumps(objectref) return ray_client_pb2.PutResponse( ref=make_remote_ref(objectref.binary(), pickled_ref))
def PutObject(self, request, context=None): obj = cloudpickle.loads(request.data) objectref = ray.put(obj) self.object_refs[objectref.binary()] = objectref logger.info("put: %s" % objectref) return ray_client_pb2.PutResponse(id=objectref.binary())
def PutObject(self, request, context=None) -> ray_client_pb2.PutResponse: obj = cloudpickle.loads(request.data) objectref = self._put_and_retain_obj(obj) return ray_client_pb2.PutResponse(id=objectref.binary())