def get_cluster_info(self, type: ray_client_pb2.ClusterInfoType.TypeEnum): req = ray_client_pb2.ClusterInfoRequest() req.type = type resp = self.server.ClusterInfo(req) if resp.WhichOneof("response_type") == "resource_table": return resp.resource_table.table return json.loads(resp.json)
def get_cluster_info(self, type: ray_client_pb2.ClusterInfoType.TypeEnum): req = ray_client_pb2.ClusterInfoRequest() req.type = type resp = self.server.ClusterInfo(req) if resp.WhichOneof("response_type") == "resource_table": # translate from a proto map to a python dict output_dict = {k: v for k, v in resp.resource_table.table.items()} return output_dict return json.loads(resp.json)
def get_cluster_info(self, type: ray_client_pb2.ClusterInfoType.TypeEnum, timeout: Optional[float] = None): req = ray_client_pb2.ClusterInfoRequest() req.type = type resp = self.server.ClusterInfo( req, timeout=timeout, metadata=self.metadata) if resp.WhichOneof("response_type") == "resource_table": # translate from a proto map to a python dict output_dict = {k: v for k, v in resp.resource_table.table.items()} return output_dict elif resp.WhichOneof("response_type") == "runtime_context": return resp.runtime_context return json.loads(resp.json)