def KVPut(self, request, context=None) -> ray_client_pb2.KVPutResponse: """Proxies internal_kv.put. This is used by the working_dir code to upload to the GCS before ray.init is called. In that case (if we don't have a server yet) we directly make the internal KV call from the proxier. Otherwise, we proxy the call to the downstream server as usual. """ if self._has_channel_for_request(context): return self._call_inner_function(request, context, "KVPut") with disable_client_hook(): already_exists = ray.experimental.internal_kv._internal_kv_put( request.key, request.value, overwrite=request.overwrite) return ray_client_pb2.KVPutResponse(already_exists=already_exists)
def KVPut(self, request, context=None) -> ray_client_pb2.KVPutResponse: with disable_client_hook(): already_exists = ray.experimental.internal_kv._internal_kv_put( request.key, request.value, overwrite=request.overwrite) return ray_client_pb2.KVPutResponse(already_exists=already_exists)