Fixes get_node_for_wire by initializing self.node Signed-off-by: Syed Tousif Ahmed <stahmed@seas.upenn.edu>
diff --git a/prjxray/node_model.py b/prjxray/node_model.py index 879f1cb..8a1b5ca 100644 --- a/prjxray/node_model.py +++ b/prjxray/node_model.py
@@ -45,7 +45,7 @@ self.nodes = None - self.wire_to_node_map = None + self.wire_to_node_map = {} if progressbar is None: self.progressbar = lambda x: x @@ -143,6 +143,9 @@ return self.nodes[tile, wire] def _build_wire_to_node_map(self): + if self.nodes is None: + self._build_nodes() + for node, wires in self.nodes.items(): for tile_wire in wires: assert tile_wire not in self.wire_to_node_map @@ -150,7 +153,7 @@ def get_node_for_wire(self, tile, wire): """ Get node for specified tile and wire. """ - if self.wire_to_node_map is None: + if not self.wire_to_node_map: self._build_wire_to_node_map() return self.wire_to_node_map[tile, wire]