The problem is that I had accidentally swapped the fig_size arguments. That line should read figsize=(n_cols + 1, n_rows + 1),. Doing this fixes the aspect ratio issue:
figsize=(n_cols + 1, n_rows + 1),