From 7f7d9d9a2c277358a54b333dd798fa1c7d4afdf2 Mon Sep 17 00:00:00 2001 From: li126com Date: Wed, 29 Nov 2023 12:05:06 +0000 Subject: [PATCH] assign rank ali --- assign_rank.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/assign_rank.py b/assign_rank.py index e2bd944..618cce3 100644 --- a/assign_rank.py +++ b/assign_rank.py @@ -150,13 +150,9 @@ def sort_node_id_list(node_id_list, node_switch_dict): def get_info_for_rank_assign(access_key_id, access_key_secret, endpoint, protocol): - print('begin2', flush=True) node_switch_dict = torch.load('node_switch_dict.pt') - print('finish read_excel', flush=True) node_id_dict, node_id_list = get_sort_info(access_key_id, access_key_secret, endpoint, protocol) - print('finish get_sort_info', flush=True) sorted_node_id_list = sort_node_id_list(node_id_list, node_switch_dict) - print('finish sort_node_id_list', flush=True) - + return node_id_dict, sorted_node_id_list \ No newline at end of file