Sort branches by completion

nickdrozd · Dec 1, 2023 · fe7f869 · fe7f869
1 parent 8b2152c
commit fe7f869
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tm/tree.py b/tm/tree.py
@@ -126,7 +126,13 @@ def distribute_branches(branches: list[Prog]) -> list[list[Prog]]:
 
     for i in range(cpus):
         end = start + size + (1 if i < rem else 0)
-        result.append(branches[start:end])
+        result.append(
+            sorted(
+                branches[ start : end ],
+                key = lambda branch: branch.count('...'),
+                reverse = True,
+            )
+        )
         start = end
 
     return result