Set team subranks for seeded draw generator

TabbycatDebate · Jan 14, 2025 · 68b1882 · 68b1882
1 parent 2265ea3
commit 68b1882
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/tabbycat/draw/generator/graph.py b/tabbycat/draw/generator/graph.py
@@ -56,7 +56,7 @@ def generate_pairings(self, brackets):
         for j, (points, teams) in enumerate(brackets.items()):
             pairings[points] = []
             graph = nx.Graph()
-            n_teams = len(teams)
+            n_teams = max([t.subrank for t in teams if t.subrank is not None])  # len(teams)
             for t1 in teams:
                 for t2 in teams:
                     penalty = self.assignment_cost(t1, t2, n_teams, j)

diff --git a/tabbycat/draw/manager.py b/tabbycat/draw/manager.py
@@ -320,6 +320,7 @@ def get_teams(self) -> Tuple[List['Team'], List['Team']]:
 
         for team in teams:
             team.points = 0
+            team.subrank = team.seed + 1
 
         return teams, byes