Function _find_reduce_close

benchmarking/insert_proton_records.py:167–178

Find the closing line of the last tt.reduce block. Returns the index of the `}) : ...` line after tt.reduce.return.

(lines: list[str])

def _find_reduce_close(lines: list[str]) -> int | None:
    """Find the closing line of the last tt.reduce block.

    Returns the index of the `}) : ...` line after tt.reduce.return.
    """
    last_reduce_return = _find_last(lines, "tt.reduce.return")
    if last_reduce_return is None:
        return None
    for i in range(last_reduce_return + 1, len(lines)):
        if re.search(r"\}\)", lines[i]):
            return i
    return None


def _find_first(lines: list[str], pattern: str) -> int | None: