Stop using recursive_tree_part

mggg · Jan 13, 2022 · 9e5f50b · 9e5f50b
1 parent 7f835db
commit 9e5f50b
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/gerrychain/proposals/tree_proposals.py b/gerrychain/proposals/tree_proposals.py
@@ -45,15 +45,16 @@ def recom(
         partition.parts[parts_to_merge[0]] | partition.parts[parts_to_merge[1]]
     )
 
-    flips = recursive_tree_part(
-        subgraph,
-        parts_to_merge,
-        pop_col=pop_col,
-        pop_target=pop_target,
-        epsilon=epsilon,
-        node_repeats=node_repeats,
-        method=method,
-    )
+    flips = {
+        x: parts_to_merge[0] 
+        for x in bipartition_tree_retworkx(
+            subgraph,
+            pop_col=pop_col,
+            pop_target=pop_target,
+            epsilon=epsilon,
+            node_repeats=node_repeats,
+        )
+    }
 
     return partition.flip(flips)