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

…eometric into rebase-txt2kg
pyg-team · Jan 10, 2025 · 4d71797 · 4d71797
2 parents c207009 + 4b1b06b
commit 4d71797
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_geometric/utils/rag/feature_store.py b/torch_geometric/utils/rag/feature_store.py
@@ -91,8 +91,8 @@ def load_subgraph(
         edge_attr = self.edge_attr[edge_id]
         # do not have access to original edge_index
         # must fill outside this function
-        return Data(x=x, edge_attr=edge_attr,
-                    node_idx=node_id, edge_idx=edge_id)
+        return Data(x=x, edge_attr=edge_attr, node_idx=node_id,
+                    edge_idx=edge_id)
 
 
 # TODO: Refactor because composition >> inheritance