diff --git a/torch_geometric/utils/rag/backend_utils.py b/torch_geometric/utils/rag/backend_utils.py index f9b912a91fba..7b9ada1bfcd3 100644 --- a/torch_geometric/utils/rag/backend_utils.py +++ b/torch_geometric/utils/rag/backend_utils.py @@ -215,7 +215,7 @@ class to use. Defaults to LocalFeatureStore. indexer = LargeGraphIndexer.from_triplets(triplets, pre_transform=pre_transform) - print("indexer._nodes[:100]=", indexer._nodes[:100]) + print("indexer._nodes[:100]=", [(indexer._nodes[key], key) for key in list(indexer._nodes.keys())[:100]]) node_feats = node_model(indexer.get_unique_node_features(), **node_method_kwargs) indexer.add_node_feature('x', node_feats)