Пример #1
0
    def KVList(self, request, context=None) -> ray_client_pb2.KVListResponse:
        """Proxies internal_kv.list.

        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, "KVList")

        with disable_client_hook():
            keys = ray.experimental.internal_kv._internal_kv_list(request.prefix)
        return ray_client_pb2.KVListResponse(keys=keys)
Пример #2
0
 def KVList(self, request, context=None) -> ray_client_pb2.KVListResponse:
     with disable_client_hook():
         keys = ray.experimental.internal_kv._internal_kv_list(
             request.prefix)
     return ray_client_pb2.KVListResponse(keys=keys)