debug

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