[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
pyg-team · Jan 10, 2025 · 4b1b06b · 4b1b06b
1 parent 3acba5e
commit 4b1b06b
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