From e6057a822728d4a7bec8404fcf59d9c914ebd89d Mon Sep 17 00:00:00 2001 From: riship Date: Fri, 10 Jan 2025 10:34:50 -0800 Subject: [PATCH] debug --- torch_geometric/utils/rag/feature_store.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/torch_geometric/utils/rag/feature_store.py b/torch_geometric/utils/rag/feature_store.py index f36b885d9fea..0dfd3918bd0f 100644 --- 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]