Minor fixes

This commit is contained in:
2024-12-01 15:21:55 +01:00
parent 16092fe5b2
commit 26ee876f7a
3 changed files with 9 additions and 7 deletions

View File

@@ -22,11 +22,11 @@ from aoc import days
@click.argument("day", required=True)
def main(day: int, timing: bool, data: IO[str]) -> None:
runner_class = days.get_runner(day)
data = data.read()
contents = data.read()
start = time.perf_counter_ns()
part1, part2 = runner_class.run_both(data)
part1, part2 = runner_class.run_both(contents)
if timing:
elapsed = time.perf_counter_ns() - start

View File

@@ -10,8 +10,7 @@ from . import CombinedRunner
class DayRunner(CombinedRunner):
@classmethod
def run_both(cls, data: str) -> tuple[Any, Any]:
data = StringIO(data)
nums = numpy.loadtxt(data, dtype=numpy.int32)
nums = numpy.loadtxt(StringIO(data), dtype=numpy.int32)
left = nums[..., 0]
right = nums[..., 1]
@@ -21,7 +20,7 @@ class DayRunner(CombinedRunner):
diff = numpy.abs(left - right).sum()
counts = defaultdict(int)
counts: defaultdict[int, int] = defaultdict(int)
for val in right:
counts[val] += 1