Merge branch 'rebase-txt2kg' of https://github.com/pyg-team/pytorch_g…

…eometric into rebase-txt2kg
pyg-team · Jan 9, 2025 · 6960715 · 6960715
2 parents b1ff4b7 + 8074555
commit 6960715
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/llm/hotpot_qa.py b/examples/llm/hotpot_qa.py
@@ -119,8 +119,8 @@
     # number of hops for neighborsampling
     num_hops = 2
     query_loader = RAGQueryLoader(
-        data=(fs, gs), seed_nodes_kwargs={"k_nodes":
-                                          knn_neighbsample_bs}, seed_edges_kwargs={"k_edges": knn_neighsample_bs},
+        data=(fs, gs), seed_nodes_kwargs={"k_nodes": knn_neighbsample_bs},
+        seed_edges_kwargs={"k_edges": knn_neighsample_bs},
         sampler_kwargs={"num_neighbors": [fanout] * num_hops},
         local_filter=make_pcst_filter(triples, model))
     """