fix: typing

Author: mbrickerd

.pre-commit-config.yaml

@@ -7,11 +7,19 @@ repos:
       - id: ruff-format
 
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.8.0
+    rev: v1.18.2
     hooks:
       - id: mypy
+        args: ["--config-file=pyproject.toml"]
         additional_dependencies:
-          ["types-requests>=2.32.0", "types-beautifulsoup4>=4.12.0"]
+          - "numpy"
+          - "scipy>=1.16.3"
+          - "scipy-stubs>=1.16.3.3"
+          - "loguru"
+          - "loguru-mypy"
+          - "rustworkx>=0.17.1"
+          - "types-requests>=2.32.0"
+          - "types-beautifulsoup4>=4.12.0"
 
   - repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v4.5.0

_2025/solutions/day10.py

@@ -14,9 +14,10 @@
 from collections import deque
 import math
 import re
-from typing import ClassVar
+from typing import ClassVar, cast
 
-from numpy import transpose
+import numpy as np
+import numpy.typing as npt
 from scipy.optimize import linprog
 
 from aoc.models.base import SolutionBase
@@ -193,16 +194,17 @@ def min_presses_for_machine(
             if any(t != 0 for t in target):
                 err_msg = f"Unreachable target {target} with given buttons"
                 raise ValueError(err_msg)
+
             return 0
 
         # Objective: minimize total button presses
         c = [1] * N
 
         # Build equality constraints: sum(button_vectors * presses) = target
-        A_eq = [self.button_to_vector(btn, num_jolt) for btn in buttons]  # noqa: N806
-        A_eq = transpose(A_eq)  # noqa: N806
-        b_eq = target
-        integrality = [1] * N
+        A_eq_list = [self.button_to_vector(btn, num_jolt) for btn in buttons]  # noqa: N806
+        A_eq: npt.NDArray[np.int_] = np.transpose(A_eq_list)  # noqa: N806
+        b_eq = np.array(target, dtype=int)
+        integrality = np.ones(N, dtype=int)
 
         res = linprog(
             c,
@@ -215,7 +217,9 @@ def min_presses_for_machine(
             err_msg = f"Unreachable target {target} with given buttons"
             raise ValueError(err_msg)
 
-        return int(math.ceil(sum(res.x)))
+        x = cast(npt.NDArray[np.float64], res.x)
+        total_presses = float(x.sum())
+        return int(math.ceil(total_presses))
 
     def part1(self, data: list[str]) -> int:
         """Sum minimum button presses to match indicator lights for all machines.

pyproject.toml

@@ -22,6 +22,7 @@ dev = [
     "types-beautifulsoup4>=4.12.0,<5.0.0",
     "pytest>=8.3.4,<9",
     "pre-commit>=4.5.0",
+    "loguru-mypy>=0.0.4",
 ]
 
 [tool.hatch.build.targets.sdist]
@@ -116,6 +117,8 @@ check_untyped_defs = true
 disallow_untyped_decorators = true
 no_implicit_optional = true
 strict_optional = true
+explicit_package_bases = true
+mypy_path = "."
 
 [[tool.mypy.overrides]]
 module = [