Re-enable lookup cache

mggg · Feb 15, 2022 · 03f11ae · 03f11ae
1 parent 0738872
commit 03f11ae
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/gerrychain/graph/graph.py b/gerrychain/graph/graph.py
@@ -402,7 +402,6 @@ def edge_indicies(self):
     def degree(self, n):
         return self.graph.degree(n)
 
-    # @functools.lru_cache(65536) # memory leak
     def lookup(self, node, field):
         return self.graph.nodes[node][field]
 
@@ -418,4 +417,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