Re-enable lookup cache

mggg · Feb 15, 2022 · fb5a961 · fb5a961
1 parent 0738872
commit fb5a961
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gerrychain/graph/graph.py b/gerrychain/graph/graph.py
@@ -402,7 +402,7 @@ def edge_indicies(self):
     def degree(self, n):
         return self.graph.degree(n)
 
-    # @functools.lru_cache(65536) # memory leak
+    @functools.lru_cache(65536)
     def lookup(self, node, field):
         return self.graph.nodes[node][field]
 
@@ -418,4 +418,4 @@ def pygraph_pop_lookup(self, field: str):
         attrs = [0] * len(self.pygraph.node_indexes())
         for node in self.pygraph.node_indexes():
             attrs[node] = float(self.pygraph[node][field])
-        return attrs
+        return attrs