From e7ac4a1d68ab76fc472323e90da90650906d009b Mon Sep 17 00:00:00 2001 From: Xiang Song Date: Tue, 10 Oct 2023 10:03:34 -0700 Subject: [PATCH] Update --- python/graphstorm/inference/ep_infer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/graphstorm/inference/ep_infer.py b/python/graphstorm/inference/ep_infer.py index 1c6c426044..d2d367fb24 100644 --- a/python/graphstorm/inference/ep_infer.py +++ b/python/graphstorm/inference/ep_infer.py @@ -141,7 +141,7 @@ def infer(self, loader, save_embed_path, save_prediction_path=None, target_ntypes.add(etype[0]) target_ntypes.add(etype[2]) # Only init the nid_shuffler when there is a node_id_mapping_file. - nid_shuffler = NodeIDShuffler(g, node_id_mapping_file, target_ntypes) \ + nid_shuffler = NodeIDShuffler(g, node_id_mapping_file, list(target_ntypes)) \ if node_id_mapping_file else None shuffled_preds = {} for etype, pred in preds.items():