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

for more information, see https://pre-commit.ci
pyg-team · Jan 14, 2025 · 5a32171 · 5a32171
1 parent 9747aa2
commit 5a32171
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/torch_geometric/loader/rag_loader.py b/torch_geometric/loader/rag_loader.py
@@ -104,7 +104,13 @@ def query(self, query: Any) -> Data:
 
         data = self.feature_store.load_subgraph(sample=subgraph_sample,
                                                 **self.loader_kwargs)
-        data.node_idx = torch.tensor(list(dict.fromkeys(list(self.graph_store.edge_index[:, data.edge_idx].t().reshape(-1)))))
+        data.node_idx = torch.tensor(
+            list(
+                dict.fromkeys(
+                    list(
+                        self.graph_store.edge_index[:,
+                                                    data.edge_idx].t().reshape(
+                                                        -1)))))
         data.x = self.feature_store.x[data.node_idx]
         if self.local_filter:
             data = self.local_filter(data, query)