feat(ai): add holes calculation

src/ai/moves/height.py

@@ -1,9 +1,7 @@
-from typing import Any
-
 import numpy as np
 
 
-def aggregate_height(field: np.ndarray[int, Any]) -> int:
+def aggregate_height(field: np.ndarray[int, np.dtype[np.uint8]]) -> int:
     """
     Calculates the aggregate height of the field.
 
@@ -13,4 +11,4 @@ def aggregate_height(field: np.ndarray[int, Any]) -> int:
     Returns:
         The aggregate height of the field.
     """
-    return np.sum(field.shape[0] - np.argmax(field, axis=0))
+    return int(np.sum(field.shape[0] - np.argmax(field, axis=0)))

src/ai/moves/holes.py

@@ -0,0 +1,26 @@
+from typing import Optional
+
+import numpy as np
+
+
+def holes(
+    field: np.ndarray[int, np.dtype[np.uint8]],
+) -> int:
+    """
+    Calculate the number of holes in each column of the given field.
+
+    Args:
+        field: The signal field.
+        peaks: Array containing peak indices. If not provided, it will be computed from the field.
+
+    Returns:
+        The total number of holes in the field.
+    """
+
+    first_nonzero_indices = np.argmax(field != 0, axis=0)
+
+    mask = (field == 0) & (
+        np.arange(field.shape[0])[:, np.newaxis] > first_nonzero_indices
+    )
+
+    return int(np.sum(mask))

src/ai/moves/lines.py

@@ -1,9 +1,7 @@
-from typing import Any
-
 import numpy as np
 
 
-def complete_lines(field: np.ndarray[int, Any]) -> int:
+def complete_lines(field: np.ndarray[int, np.dtype[np.uint8]]) -> int:
     """
     Calculates the number of complete lines in the field.
 
@@ -13,4 +11,4 @@ def complete_lines(field: np.ndarray[int, Any]) -> int:
     Returns:
         The number of complete lines in the field.
     """
-    return np.sum(np.all(field, axis=1))
+    return int(np.sum(np.all(field, axis=1)))

tests/ai/test_moves.py

@@ -2,6 +2,7 @@ import unittest
 
 import numpy as np
 from ai.moves.height import aggregate_height
+from ai.moves.holes import holes
 from ai.moves.lines import complete_lines
 
 
@@ -23,3 +24,6 @@ class TestFitness(unittest.TestCase):
 
     def test_complete_lines(self) -> None:
         self.assertEqual(complete_lines(self.field), 2)
+
+    def test_holes(self) -> None:
+        self.assertEqual(holes(self.field), 2)