debug

pyg-team · Jan 10, 2025 · f959254 · f959254
2 parents f0d3710 + 0d3318b
commit f959254
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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)