Partition: avoid tasks too close in time

repartir_taches/entrypoint.py

@@ -75,6 +75,7 @@ def assigner_taches(root_task: Category | Task, group_count: int):
         multiplicity[t_id] = task.nb_groups
     repart: list[list[TaskId]] = partition(
         bin_count=group_count,
+        tasks=all_tasks,
         costs=costs,
         multiplicity=multiplicity,
     )

repartir_taches/partition.py

@@ -1,12 +1,16 @@
 """ Implements Multiway number partitioning greedy algorithm """
 
 import typing as t
+import logging
 from sortedcontainers import SortedList
+from .config import Task
 
 __all__ = ["TaskId", "partition"]
 
 TaskId = t.NewType("TaskId", int)
 
+logger = logging.getLogger(__name__)
+
 
 class PartitionException(Exception):
     """An exception occurring during partitioning"""
@@ -36,7 +40,10 @@ class Bin:
 
 
 def partition(
-    bin_count: int, costs: dict[TaskId, int], multiplicity: dict[TaskId, int]
+    bin_count: int,
+    tasks: list[Task],
+    costs: dict[TaskId, int],
+    multiplicity: dict[TaskId, int],
 ) -> list[list[TaskId]]:
     """Partitions the tasks, each with cost `costs[i]`, into `bin_count` bins. Each
     task has multiplicity `multiplicity[i]`, copies of the same task being mutually
@@ -50,22 +57,36 @@ def partition(
     ordered_tasks.sort(key=lambda x: costs[x], reverse=True)
 
     for task in ordered_tasks:
-        least_full: Bin
+        possible_bins: list[int] = []
-        least_full_pos: int
+        min_possible: t.Optional[int] = None
         for pos, cur_bin in enumerate(bins):
-            if task not in cur_bin:
+            if min_possible is not None and cur_bin.cost > min_possible:
-                least_full = cur_bin
-                least_full_pos = pos
                 break
-        else:
+            if task not in cur_bin:
+                if min_possible is None:
+                    min_possible = cur_bin.cost
+                possible_bins.append(pos)
+        if not possible_bins:
             raise UnsolvableConflict(
                 "Pas assez de groupes pour affecter la tâche "
-                + f"{task} {multiplicity[task]} fois."
+                + f"{tasks[task].qualified_name} {multiplicity[task]} fois."
             )
 
-        del bins[least_full_pos]
+        # Pick one of the groups -- maximize distance to other tasks
-        least_full.add(task, costs[task])
+        closest_assigned = []
-        bins.add(least_full)
+        for cur_bin_id in possible_bins:
+            cur_bin = bins[cur_bin_id]
+            if cur_bin.elts:
+                closest_assigned.append(min((abs(elt - task) for elt in cur_bin.elts)))
+            else:
+                closest_assigned.append(len(costs) + 1)
+
+        assign_to_bin_id, _ = max(enumerate(closest_assigned), key=lambda x: x[1])
+        assign_to_bin = bins[possible_bins[assign_to_bin_id]]
+
+        del bins[assign_to_bin_id]
+        assign_to_bin.add(task, costs[task])
+        bins.add(assign_to_bin)
 
     out: list[list[TaskId]] = []
     for cur_bin in bins: