Skip to content

GitLab

Unverified Commit 394794b1 authored by xiang song(charlie.song)'s avatar xiang song(charlie.song) Committed by GitHub
Browse files

Fix bug when feature_tid is empty (#4985) 


Co-authored-by: default avatarXiang Song <xiangsx@amazon.com>
parent 279e2e8f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/distpartitioning/data_shuffle.py
View file @ 394794b1
...@@ -431,7 +431,7 @@ def exchange_features(rank, world_size, num_parts, feature_tids, type_id_map, id ...@@ -431,7 +431,7 @@ def exchange_features(rank, world_size, num_parts, feature_tids, type_id_map, id
own_global_ids) own_global_ids)
end = timer() end = timer()
logging.info(f'[Rank: {rank}] Total time for feature exchange {feat_key}: {timedelta(seconds = end - start)}') logging.info(f'[Rank: {rank}] Total time for feature exchange: {timedelta(seconds = end - start)}')
return own_features, own_global_ids return own_features, own_global_ids
def exchange_graph_data(rank, world_size, num_parts, node_features, edge_features, def exchange_graph_data(rank, world_size, num_parts, node_features, edge_features,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment