Skip to content

Commit 3a3c160

Browse files
committed
Make node options explicit
1 parent 9f9b616 commit 3a3c160

1 file changed

Lines changed: 9 additions & 2 deletions

File tree

task_queue/grid_optimizer.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -600,7 +600,7 @@ def _add_node(self, label, **kwargs):
600600
"how_added": "automatic",
601601
"type_fixed": False,
602602
"cluster_label": 0,
603-
"n_connection_links": "0",
603+
"n_connection_links": 0,
604604
"n_distribution_links": 0,
605605
"parent": "unknown",
606606
"distribution_cost": 0,
@@ -1683,10 +1683,12 @@ def kmeans_clustering(self, n_clusters: int, consumer_indices: pd.Index = None):
16831683
poles["longitude"] = 0
16841684
poles["distance_to_load_center"] = 0
16851685
poles["type_fixed"] = False
1686-
poles["n_connection_links"] = "0"
1686+
poles["n_connection_links"] = 0
16871687
poles["n_distribution_links"] = 0
16881688
poles["parent"] = "unknown"
16891689
poles["distribution_cost"] = 0
1690+
poles["custom_specification"] = ""
1691+
poles["shs_options"] = 0
16901692
self.nodes = pd.concat([self.nodes, poles])
16911693
self.nodes.index = self.nodes.index.astype("str")
16921694
self.convert_lonlat_xy(inverse=True)
@@ -1735,6 +1737,11 @@ def _add_pole(x, y):
17351737
poles["how_added"] = "road-sampled"
17361738
poles["type_fixed"] = False
17371739
poles["is_connected"] = True
1740+
poles["n_connection_links"] = 0
1741+
poles["n_distribution_links"] = 0
1742+
poles["parent"] = "unknown"
1743+
poles["custom_specification"] = ""
1744+
poles["shs_options"] = 0
17381745
# High-offset cluster labels avoid collision with k-means labels (0..n)
17391746
poles["cluster_label"] = range(self._ROAD_POLE_CLUSTER_OFFSET, self._ROAD_POLE_CLUSTER_OFFSET + len(poles))
17401747

0 commit comments

Comments
 (0)