Skip to content

Commit 2e82372

Browse files
committed
Fix ruff
1 parent 02a5b31 commit 2e82372

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

modules/transforms/liftings/graph2simplicial/line_lifting.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,9 @@
22
import torch_geometric
33
from toponetx.classes import SimplicialComplex
44

5-
from modules.transforms.liftings.graph2simplicial.base import Graph2SimplicialLifting
5+
from modules.transforms.liftings.graph2simplicial.base import (
6+
Graph2SimplicialLifting,
7+
)
68

79

810
class SimplicialLineLifting(Graph2SimplicialLifting):
@@ -47,11 +49,15 @@ def lift_topology(self, data: torch_geometric.data.Data) -> dict:
4749
simplices = list(cliques) # list(map(lambda x: set(x), cliques))
4850

4951
# we need to rename simplices here since now vertices are named as pairs
50-
self.rename_vertices_dict = {node: i for i, node in enumerate(line_graph.nodes)}
52+
self.rename_vertices_dict = {
53+
node: i for i, node in enumerate(line_graph.nodes)
54+
}
5155
self.rename_vertices_dict_inverse = {
5256
i: node for node, i in self.rename_vertices_dict.items()
5357
}
54-
renamed_line_graph = nx.relabel_nodes(line_graph, self.rename_vertices_dict)
58+
renamed_line_graph = nx.relabel_nodes(
59+
line_graph, self.rename_vertices_dict
60+
)
5561

5662
renamed_simplices = [
5763
{self.rename_vertices_dict[vertex] for vertex in simplex}
@@ -70,4 +76,6 @@ def lift_topology(self, data: torch_geometric.data.Data) -> dict:
7076
renamed_node_features, name="features"
7177
)
7278

73-
return self._get_lifted_topology(simplicial_complex, renamed_line_graph)
79+
return self._get_lifted_topology(
80+
simplicial_complex, renamed_line_graph
81+
)

0 commit comments

Comments
 (0)