debug

pyg-team · Jan 10, 2025 · 454bf0a · 454bf0a
1 parent 1d3030e
commit 454bf0a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torch_geometric/utils/rag/feature_store.py b/torch_geometric/utils/rag/feature_store.py
@@ -86,8 +86,9 @@ def load_subgraph(
         # NOTE: torch_geometric.loader.utils.filter_custom_store can be used
         # here if it supported edge features
         node_id = sample.node
+        print("sampled node id in load subgraph=", node_id)
         edge_id = sample.edge
-        edge_index = torch.stack((sample.row, sample.col), dim=0)
+        edge_index = torch.stack((sample.orig_row, sample.orig_col), dim=0)
         x = self.x[node_id]
         edge_attr = self.edge_attr[edge_id]