Update county splits implementation to use cached graph attrs

mggg · Feb 15, 2022 · a97beed · a97beed
1 parent 989c9c5
commit a97beed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gerrychain/updaters/county_splits.py b/gerrychain/updaters/county_splits.py
@@ -35,8 +35,8 @@ def compute_county_splits(partition, county_field, partition_field):
     if not partition.parent:
         county_dict = dict()
 
-        for node in partition.graph.nodes:
-            county = partition.graph.nodes[node][county_field]
+        for node in partition.graph.node_indices:
+            county = partition.graph.lookup(node, county_field)
             if county in county_dict:
                 split, nodes, seen = county_dict[county]
             else: