Speedup max repeating player rule

DimaKudosh · Aug 19, 2019 · 63be0a3 · 63be0a3
1 parent 42ff34a
commit 63be0a3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pydfs_lineup_optimizer/rules.py b/pydfs_lineup_optimizer/rules.py
@@ -268,8 +268,7 @@ def apply_for_iteration(self, solver, players_dict, result):
         max_repeating_players = self.optimizer.max_repeating_players
         if max_repeating_players is None or not result:
             return
-        for players_combination in combinations(result.players, max_repeating_players + 1):
-            self.exclude_combinations.append([players_dict[player] for player in players_combination])
+        self.exclude_combinations.append([players_dict[player] for player in result])
         for players_combination in self.exclude_combinations:
             solver.add_constraint(players_combination, None, SolverSign.LTE, max_repeating_players)