def get_node_info(node_config): if not node_config: Printer.print_error('No node was found.') sys.exit(1) node = node_config.to_dict() node.pop('gpus') Printer.print_header("Node info:") dict_tabulate(Printer.add_memory_unit(node, 'memory')) gpus_items = list_dicts_to_tabulate([Printer.add_memory_unit(gpu.to_light_dict(), 'memory') for gpu in node_config.gpus]) if gpus_items: Printer.print_header("Node GPUs:") dict_tabulate(gpus_items, is_list_dict=True)
def get_cluster_info(cluster_config): Printer.print_header("Cluster info:") cluster_dict = cluster_config.to_light_dict() cluster_info = cluster_dict['version_api'] dict_tabulate(cluster_info) nodes = list_dicts_to_tabulate([Printer.add_memory_unit(node.to_light_dict(), 'memory') for node in cluster_config.nodes]) if nodes: Printer.print_header("Cluster Nodes:") nodes.pop('status', None) nodes.pop('kernel_version', None) dict_tabulate(nodes, is_list_dict=True)