speedup indexing

pyg-team · Jan 10, 2025 · 725cab0 · 725cab0
1 parent 3e0a0de
commit 725cab0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch_geometric/utils/rag/backend_utils.py b/torch_geometric/utils/rag/backend_utils.py
@@ -217,6 +217,7 @@ class to use. Defaults to LocalFeatureStore.
                                               pre_transform=pre_transform)
 
     node_feats = node_model(indexer.get_node_features(), **node_method_kwargs)
+    print("node_feats in create_remote_backend=", node_feats)
     indexer.add_node_feature('x', node_feats)
 
     edge_feats = edge_model(