Skip to content

Commit 75e0633

Browse files
added add_edge_cells to the preprocessing function to fix error when exporting data from partitioner
1 parent 0c5dded commit 75e0633

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

superblockify/utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
from .config import logger, Config
3030
from .graph_stats import basic_graph_stats
3131
from .population.approximation import add_edge_population
32+
from .population.tessellation import add_edge_cells
3233

3334

3435
def extract_attributes(graph, edge_attributes, node_attributes):
@@ -141,6 +142,9 @@ def preprocess_graph(G: MultiDiGraph, boundary_buffer_dist: float = 200) -> Mult
141142
# Adding the edge population data
142143
add_edge_population(G)
143144

145+
# Adding tessellation cells to edges
146+
add_edge_cells(G)
147+
144148
return G
145149

146150

0 commit comments

Comments
 (0)