From 5a3217195f24f37b8a7b178906cee0b56d7c312b Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 14 Jan 2025 03:57:59 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- torch_geometric/loader/rag_loader.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/torch_geometric/loader/rag_loader.py b/torch_geometric/loader/rag_loader.py index f6e6dbbba64f..1758b61760d0 100644 --- 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)