Merge branch 'to_hetero_static' of github.com:pyg-team/pytorch_geomet…

…ric into to_hetero_static
pyg-team · Oct 22, 2023 · 6233437 · 6233437
2 parents 0518c1e + 8aec3c6
commit 6233437
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/torch_geometric/transforms/add_positional_encoding.py b/torch_geometric/transforms/add_positional_encoding.py
@@ -147,8 +147,6 @@ def forward(self, data: Data) -> Data:
         pe_list = [get_self_loop_attr(*to_edge_index(out), num_nodes=N)]
         for _ in range(self.walk_length - 1):
             out = out @ adj
-            if out.layout == torch.sparse_coo:
-                out = out._coalesced_(True)
             pe_list.append(get_self_loop_attr(*to_edge_index(out), N))
         pe = torch.stack(pe_list, dim=-1)