Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jan 10, 2025
1 parent 3841de2 commit 9c2850b
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions torch_geometric/utils/rag/feature_store.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,9 +88,11 @@ def load_subgraph(
node_id = sample.node
print("sampled node id in load subgraph=", node_id)
print("sampled src ids in load subgraph=", sample.row)
print("num src ids not in node ids inside `load_subgraph`=",sum([src_id not in node_id for src_id in sample.row]))
print("num src ids not in node ids inside `load_subgraph`=",
sum([src_id not in node_id for src_id in sample.row]))
print("sampled dst ids in load subgraph=", sample.col)
print("num dst ids not in node ids inside `load_subgraph`=",sum([dst_id not in node_id for dst_id in sample.col]))
print("num dst ids not in node ids inside `load_subgraph`=",
sum([dst_id not in node_id for dst_id in sample.col]))
edge_id = sample.edge
edge_index = torch.stack((sample.row, sample.col), dim=0)
x = self.x[node_id]
Expand Down

0 comments on commit 9c2850b

Please sign in to comment.