Richie/dotfiles
1
0
Fork 0
mirror of https://github.com/RichieCahill/dotfiles.git synced 2026-04-17 04:58:19 -04:00
Code Issues Packages Projects Releases Wiki Activity

speed up check_nobles_for_player

Browse Source
This commit is contained in:
Richie Cahill
2025-11-16 15:30:11 -05:00
parent 55e652a51d
commit 6ce622e93e
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
python/splendor/base.py
View File

@@ -331,6 +331,14 @@ def enforce_token_limit(
game.bank[color] += amount
def _check_nobles_for_player(player: PlayerState, noble: Noble) -> bool:
# this rule is slower
for color, cost in noble.requirements.items(): # noqa: SIM110
if player.discounts[color] < cost:
return False
return True
def check_nobles_for_player(
game: GameState,
strategy: Strategy,
@@ -340,11 +348,7 @@ def check_nobles_for_player(
if game.noble_min_requirements > max(player.discounts.values()):
return
candidates = [
noble
for noble in game.available_nobles
if all(player.discounts.get(color, 0) >= requirement for color, requirement in noble.requirements.items())
]
candidates = [noble for noble in game.available_nobles if _check_nobles_for_player(player, noble)]
if not candidates:
return