kristoferssolo/Tetris
1
0
Fork 0
mirror of https://github.com/kristoferssolo/Tetris.git synced 2025-10-21 20:00:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(ai): add get_holes_sum

Browse Source
This commit is contained in:
Kristofers Solo 2024-01-05 18:40:00 +02:00
parent 25e43aeea3
commit 01a84bc201
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/ai/fitness/holes.py
View File

@ -20,3 +20,14 @@ def get_holes(
holes[col] = np.count_nonzero(field[int(start) :, col] == 0)
return holes
def get_holes_sum(
*, holes: Optional[np.ndarray] = None, field: Optional[np.ndarray] = None
) -> int:
if holes is None and field is None:
raise ValueError("holes and field cannot both be None")
elif holes is None:
holes = get_holes(field)
return int(np.sum(holes))