Fix linting.

src/holt59/aoc/2022/day15.py

@@ -1,20 +1,22 @@
 import sys
+from typing import Any
 
 import numpy as np
-import parse
+import parse  # type: ignore
+from numpy.typing import NDArray
 
 
 def part1(sensor_to_beacon: dict[tuple[int, int], tuple[int, int]], row: int) -> int:
-    no_beacons_row_l: list[np.ndarray] = []
+    no_beacons_row_l: list[NDArray[np.floating[Any]]] = []
 
     for (sx, sy), (bx, by) in sensor_to_beacon.items():
         d = abs(sx - bx) + abs(sy - by)  # closest
 
-        no_beacons_row_l.append(sx - np.arange(0, d - abs(sy - row) + 1))
-        no_beacons_row_l.append(sx + np.arange(0, d - abs(sy - row) + 1))
+        no_beacons_row_l.append(sx - np.arange(0, d - abs(sy - row) + 1))  # type: ignore
+        no_beacons_row_l.append(sx + np.arange(0, d - abs(sy - row) + 1))  # type: ignore
 
     beacons_at_row = set(bx for (bx, by) in sensor_to_beacon.values() if by == row)
-    no_beacons_row = set(np.concatenate(no_beacons_row_l)).difference(beacons_at_row)
+    no_beacons_row = set(np.concatenate(no_beacons_row_l)).difference(beacons_at_row)  # type: ignore
 
     return len(no_beacons_row)
 
@@ -56,11 +58,12 @@ def part2_cplex(
 
     for (sx, sy), (bx, by) in sensor_to_beacon.items():
         d = abs(sx - bx) + abs(sy - by)
-        m.add_constraint(m.abs(x - sx) + m.abs(y - sy) >= d + 1, ctname=f"ct_{sx}_{sy}")
+        m.add_constraint(m.abs(x - sx) + m.abs(y - sy) >= d + 1, ctname=f"ct_{sx}_{sy}")  # type: ignore
 
     m.set_objective("min", x + y)
 
     s = m.solve()
+    assert s is not None
 
     vx = int(s.get_value(x))
     vy = int(s.get_value(y))
@@ -72,7 +75,7 @@ lines = sys.stdin.read().splitlines()
 sensor_to_beacon: dict[tuple[int, int], tuple[int, int]] = {}
 
 for line in lines:
-    r = parse.parse(
+    r: dict[str, str] = parse.parse(  # type: ignore
         "Sensor at x={sx}, y={sy}: closest beacon is at x={bx}, y={by}", line
     )
     sensor_to_beacon[int(r["sx"]), int(r["sy"])] = (int(r["bx"]), int(r["by"]))