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