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)
