boggle state transitions for an n x n board I acknowledge it's a bad idea in the first place, but should I indent this?

boggle_transitions.py

def get_transitions(n):
    """Return a dictionary of array spots to neighbors for an nxn grid"""
    return {initial_row*n + initial_col :
                {row * n + col
                 for row in range(max(0, initial_row-1), min(n, initial_row+2))
                 for col in range(max(0, initial_col-1), min(n, initial_col+2))
                 if (row, col) != (initial_row, initial_col)}
            for initial_row in range(n)
            for initial_col in range(n)}

Unsolicited refactor because I'm distractible:

def get_transitions(n):
    "Return a dictionary of array spots to neighbors for an nxn grid."
    def neighbors(row, col):
        return {at(to_row, to_col)
                for to_row in around(row)
                for to_col in around(col)
                if (to_row, to_col) != (row, col)}
    def at(row, col):
        return row * n + col
    def around(i):
        return range(max(0, i-1), min(i+2, n))
    return {at(row, col): neighbors(row, col)
            for row in range(n)
            for col in range(n)}

Nice! I was reaching for something like around, much clearer; thanks!