debug

pyg-team · Jan 10, 2025 · 21c5875 · 21c5875
1 parent 1ecdaa7
commit 21c5875
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/loader/rag_loader.py b/torch_geometric/loader/rag_loader.py
@@ -98,7 +98,7 @@ def query(self, query: Any) -> Data:
         # Graph Store does not Use These, save computation
         # seed_edges = self.feature_store.retrieve_seed_edges(
         #     query, **self.seed_edges_kwargs)
-
+        print("Seed nodes=", seed_nodes.unique())
         subgraph_sample = self.graph_store.sample_subgraph(
             seed_nodes, **self.sampler_kwargs)