Advent of code 2024 day 2

def day2(s, part2=False):
  def safe(values):
    diff = [b - a for a, b in itertools.pairwise(values)]
    return all(1 <= d <= 3 for d in diff) or all(-3 <= d <= -1 for d in diff)

  def check(values):
    return safe(values) or (
        part2 and any(safe(values[:i] + values[i + 1 :]) for i in range(len(values)))
    )

  return sum(check(list(map(int, line.split()))) for line in s.splitlines())